MLX: Array framework for Apple silicon machine learning
Lazy-evaluated NumPy-like arrays optimized for Apple silicon.
Learn more about mlx
MLX is an array computation framework designed specifically for machine learning workloads on Apple silicon processors. It implements lazy evaluation with dynamic graph construction, allowing computations to be deferred until results are needed and computation graphs to adapt to changing input shapes without recompilation. The framework features a unified memory model where arrays reside in shared memory accessible by both CPU and GPU devices without explicit data transfers. Common use cases include training transformer models, fine-tuning large language models with techniques like LoRA, and running inference for generative tasks on macOS and iOS devices.
Unified memory architecture
Arrays live in shared memory accessible by both CPU and GPU without requiring explicit data transfers between devices. Operations can transparently execute on either device type.
Dynamic graph construction
Computation graphs are built at runtime rather than pre-compiled, allowing function arguments to change shapes without triggering recompilation or requiring graph tracing.
Multi-language API consistency
Python, C++, C, and Swift APIs closely mirror each other, with higher-level packages like mlx.nn and mlx.optimizers following PyTorch conventions for model building.
import mlx.core as mx
a = mx.array([1, 2, 3, 4])
b = mx.array([5, 6, 7, 8])
c = a + b
print(c)Release notes contain only a rocket emoji with no technical details provided.
- –Release notes do not specify breaking changes, requirements, or migration steps.
- –No bug fixes, security patches, or new capabilities are documented in the provided text.
Release notes do not specify changes, breaking updates, or requirements.
- –No actionable information provided in the release notes for this version.
- –Review commit history or changelog separately to identify actual changes before upgrading.
Release notes contain only an upward arrow emoji with no technical details provided.
- –Release notes do not specify breaking changes, new features, or bug fixes.
- –Consult the project's commit history or changelog for actual changes in this version.
Related Repositories
Discover similar tools and frameworks used by developers
crawl4ai
Async browser automation extracting web content for LLMs.
mediapipe
Graph-based framework for streaming media ML pipelines.
Wan2.1
Diffusion transformer models for text and image-to-video generation.
presentation-ai
AI-powered slide generator with multi-model integration and themes.
segment-anything
Transformer-based promptable segmentation with zero-shot generalization.