Skip to documentation
Deploy Package

Models and runtimes

MNIST

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 S

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.