[adolph]

If PyTorch does the 1-2 things you need and Tinygrad doesn't do, then what are you going to use?

The Python source distribution has long maintained the philosophy of “batteries included” – having a rich and versatile standard library which is immediately available, without making the user download separate packages.

https://peps.python.org/pep-0206/

OTOH:

  Simple is better than complex.
  Complex is better than complicated.

https://peps.python.org/pep-0020/

[jononor]

PyTorch of course. Or alternatively a lib or custom code on top of TinyGrad. Is that a problem?