This is a port of the MNIST training code from the Python MLX example. This example uses a LeNet instead of an MLP.
It provides code to:
- Download the MNIST test/train data
- Build the LeNet
- Some functions to shuffle and batch the data
See mnist-tool for an example of how to run this. The training loop also lives there.