If anyone is interested in a dead-simple numpy-based autograd library: https://mygrad.readthedocs.io/en/latest/ https://mygrad.readthedocs.io/en/latest/ I crea

If anyone is interested in a dead-simple numpy-based autograd library: https://mygrad.readthedocs.io/en/latest/ https://mygrad.readthedocs.io/en/latest/ I created this for educational purposes, but it is quite robust, simple, well-tested, and well-documented. It also includes neural network style operations like N-dimensional convolutions and pooling. Plus backprop through all variations of einsum :)