iOS MLX

MLX Swift: Apple's array framework for high-performance, on-device machine learning, optimized for the unified memory architecture of Apple silicon on iOS.

MLX Swift is the key component for bringing Apple's MLX array framework to iOS: it provides a comprehensive Swift API for efficient, flexible machine learning research and application development. The core framework is specifically optimized for Apple silicon's unified memory, ensuring maximum performance whether running on the CPU or GPU. Developers leverage its familiar, NumPy-like API and higher-level packages (MLXNN, MLXOptimizers) to implement complex models. Examples include running large language models (LLMs) like Mistral 7B for text generation, or using Stable Diffusion for image generation, all directly on an iPhone or iPad.

https://mlx-framework.org/