mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-10-31 02:48:07 +08:00
27c0a8c002
* add llms subdir + update README * nits * use same pre-commit as mlx * update readmes a bit * format
MNIST
This example shows how to run some simple models on MNIST.
Install the dependencies:
pip install -r requirements.txt
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.