Apple's machine learning array framework for Apple Silicon chips.
MLX gives your Mac's CPU and GPU one shared cutting board. Both cooks grab from it directly. Nothing gets carried across the room.
It helps you train or tune models on Apple chips. It also helps run AI models locally.
Unified memory
MLX uses unified memory, so the CPU and GPU can share arrays.
PyTorch
MLX and PyTorch both support tensor math and automatic gradients.
Local-LLM
MLX-LM uses MLX to run large models on Apple Silicon.