micrograd
micrograd copied to clipboard
Vectorized implementation with PyTorch flavor
Here is a vectorized implementation with PyTorch flavor built on top of NumPy / CuPy: https://github.com/conscell/ugrad