Apple Releases Open Source MLX Framework for Efficient Machine Learning on Apple Silicon

Apple says MLX outperforms PyTorch in some scenarios while generating batches of images using Stable Diffusion on a Mac powered by an M2 Ultra chip.

Apple Releases Open Source MLX Framework for Efficient Machine Learning on Apple Silicon

Photo Credit: Apple

MLX could help developers optimise the performance of ML models on Apple's computers

  • Apple's MLX framework is designed to run on Apple Silicon hardware
  • MLX boasts a unified memory model not supported by other frameworks
  • The new MLX framework is open source and can be accessed via GitHub

Apple recently released MLX — or ML Explore — the company's machine learning (ML) framework for Apple Silicon computers. The company's latest framework is specifically designed to simplify the process of training and running ML models on computers that are powered by Apple's M1, M2, and M3 series chips. The company says that MLX features a unified memory model. Apple has also demonstrated the use of the framework, which is open source, allowing machine learning enthusiasts to run the framework on their laptop or computer.

According to details shared by Apple on code hosting platform GitHub, the MLX framework has a C++ API along with a Python API that is closely based on NumPy, the Python library for scientific computing. Users can also take advantage of higher-level packages that enable them to build and run more complex models on their computer, according to Apple.

MLX simplifies the process of training and running ML models on a computer — developers were previously forced to rely on a translator to convert and optimise their models (using CoreML). This has now been replaced by MLX, which allows users running Apple Silicon computers to train and run their models directly on their own devices.

generated mlx mlx apple

Apple shared this image of a big red sign with the text MLX, generated by Stable Diffusion in MLX
Photo Credit: GitHub/ Apple


Apple says that the MLX's design follows other popular frameworks used today, including ArrayFireJax, NumPy, and PyTorch. The firm has touted its framework's unified memory model — MLX arrays live in shared memory, while operations on them can be performed on any device types (currently, Apple supports the CPU and GPU) without the need to create copies of data.

The company has also shared examples of MLX in action, performing tasks like image generation using Stable Diffusion on Apple Silicon hardware. When generating a batch of images, Apple says that MLX is faster than PyTorch for batch sizes of 6,8,12, and 16 — with up to 40 percent higher throughput than the latter.

The tests were conducted on a Mac powered by an M2 Ultra chip, the company's fastest processor to date — MLX is capable of generating 16 images in 90 seconds, while PyTorch would take around 120 seconds to perform the same task, according to the company.

Other examples of MLX in action include generating text using Meta's open source LLaMA language model, as well as the Mistral large language model. AI and ML researchers can also use OpenAI's open source Whisper tool to run the speech recognition models on their computer using MLX.

The release of Apple's MLX framework could help make ML research and development easier on the company's hardware, eventually allowing developers to bring better tools that could be used for apps and services that offer on-device ML features running efficiently on a user's computer.

Is the Samsung Galaxy Z Flip 5 the best foldable phone you can buy in India right now? We discuss the company's new clamshell-style foldable handset on the latest episode of Orbital, the Gadgets 360 podcast. Orbital is available on Spotify, Gaana, JioSaavn, Google Podcasts, Apple Podcasts, Amazon Music and wherever you get your podcasts.
Affiliate links may be automatically generated - see our ethics statement for details.

For the latest tech news and reviews, follow Gadgets 360 on X, Facebook, WhatsApp, Threads and Google News. For the latest videos on gadgets and tech, subscribe to our YouTube channel. If you want to know everything about top influencers, follow our in-house Who'sThat360 on Instagram and YouTube.

David Delima
As a writer on technology with Gadgets 360, David Delima is interested in open-source technology, cybersecurity, consumer privacy, and loves to read and write about how the Internet works. David can be contacted via email at, on Twitter at @DxDavey, and Mastodon at More
India’s Concrete Crypto Rules Could Arrive by Mid-2025, Government Official Claims
Share on Facebook Gadgets360 Twitter Share Tweet Snapchat Share Reddit Comment google-newsGoogle News


Follow Us


© Copyright Red Pixels Ventures Limited 2024. All rights reserved.
Trending Products »
Latest Tech News »