nx

Ten Years Later: Rebuilding My 2016 Neural Network in Nx and Axon

In 2016 I hand-derived the gradient for a single neuron in NumPy. Ten years later I rebuilt the same network in Nx and Axon, and the math I sweated over is now one function call.