Files

Awni Hannun 4ed4ec69e7 LeNet on MNIST + readme update (#12 )

* LeNet on MNIST + readme update

* tanh + remove device toggle

* remove device entirely

2024-03-04 14:16:20 -08:00

MLX Swift Examples

Example MLX Swift programs.

MNISTTrainer: An example that runs on both iOS and macOS that downloads MNIST training data and trains a LeNet.
LLMEval: An example that runs on both iOS and macOS that downloads an LLM and tokenizer from Hugging Face and and generates text from a given prompt.
LinearModelTraining: An example that trains a simple linear model.
llm-tool: A command line tool for generating text using a variety of LLMs available on the Hugging Face hub.
mnist-tool: A command line tool for training a a LeNet on MNIST.