MLX vs MPS vs CUDA: a Benchmark

A first benchmark of Apple’s new ML framework MLX

Tristan Bilot
Towards Data Science
6 min readDec 15, 2023

--

Photo by Javier Allegue Barros on Unsplash

If you’re a Mac user and a deep learning enthusiast, you’ve probably wished at some point that your Mac could handle those heavy models, right? Well, guess what? Apple just released MLX, a framework for running ML models efficiently on Apple Silicon.

--

--

PhD student in GNNs for cybersecurity. Basically writing about deep learning, programming and performance 📝.