mlx-swift-examples/Libraries/MNIST/README.md

MNIST

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.