MLX Swift

Overview

MLX Swift is the Swift API for MLX, Apple's array framework for machine learning on Apple silicon. It brings MLX to Swift to make research and experiments on Apple silicon easier. Language and vision-language model implementations live in the separate mlx-swift-lm package.

Features

• Array framework for machine learning on Apple silicon
• Builds with Xcode or SwiftPM, plus CMake with a native Linux option
• Examples include MNIST training, a chat app, LLM evaluation and Stable Diffusion
• Version numbers generally match the core MLX releases

Requirements

• Apple silicon

Install

1. Add the package to Package.swift, or add https://github.com/ml-explore/mlx-swift.git in Xcode

   .package(url: "https://github.com/ml-explore/mlx-swift", from: "0.10.0")