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 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 LeNet on MNIST.
The application and command line tool examples can be run from Xcode or from the command line:
./mlx-run llm-tool --prompt "swift programming language"
See also:
The MLXLLM and MLXMNIST libraries in the example repo are available as Swift Packages.
Add the following dependency to your Package.swift
.package(url: "https://github.com/ml-explore/mlx-swift-examples/", branch: "main"),
Then add one library or both libraries to the target as a dependency.
.target(
name: "YourTargetName",
dependencies: [
.product(name: "LLM", package: "mlx-swift-examples")
]),
Alternatively, add https://github.com/ml-explore/mlx-swift-examples/
to the Project Dependencies
and set the Dependency Rule
to Branch
and main
in Xcode.