I Tested Apple’s New MLX Framework Against Torch on M2 Air
MLX vs Torch on BERT — Training, Inference, and CPU Usage Comparison
Published in
4 min readDec 22, 2023
On Tuesday, Apple’s AI team released “MLX” — the new machine learning framework designed to work specifically for the Apple Silicon Chips. The design of MLX was inspired by frameworks like NumPy, PyTorch, Jax, and ArrayFire.