zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 01:17:28 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
31bc57c4ff
mlx-examples/mnist
History
Awni Hannun 31bc57c4ff add copyright in source
2023-11-30 11:08:53 -08:00
..
jax_main.py add copyright in source 2023-11-30 11:08:53 -08:00
main.py add copyright in source 2023-11-30 11:08:53 -08:00
mnist.py add copyright in source 2023-11-30 11:08:53 -08:00
README.md a few examples 2023-11-29 08:17:26 -08:00
torch_main.py add copyright in source 2023-11-30 11:08:53 -08:00

README.md

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.