tinygrad/tinygrad — You like pytorch?

tinygrad is a minimalist deep‑learning framework that implements the core concepts of autograd and tensor operations in just a few hundred lines of Python code.