MNIST

This example shows how to run some simple models on MNIST. The only dependency is MLX.

Run the example with:

python main.py

By default the example runs on the CPU. To run on the GPU, use:

python main.py --gpu

To run the PyTorch or Jax examples install the respective framework.