tinygrad-inspired autograd library todo: mnist gpt2 diffusion models advanced todo: compile to CUDA/Metal