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.