so something like this? https://tinygrad.org/#tinybox