Right now it's just a tensor manipulation lib but will be adding an autograd engine soon. It's been fun learning about strides and doing matmuls by hand and then coding it without numpy.
1. https://github.com/keshavsaharia/numbers/blob/dev/lib/nn/neu...
2. https://github.com/keshavsaharia/numbers/blob/dev/lib/cnn/cn... (still working on the visualization)
Hope you find these useful in your own learning journey!