nn.py

The spelled-out intro to neural networks and backpropagation: building micrograd

Andrej Karpathy

The second file in micrograd that provides tiny neural network primitives (Neuron, Layer, MLP) built on top of the engine.