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.

Advertisement
Written by David Delima, Edited by Siddharth Suvarna | Updated: 8 December 2023 17:24 IST
Highlights
  • 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 Releases Open Source MLX Framework for Efficient Machine Learning on Apple Silicon

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

Photo Credit: Apple

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.

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

Advertisement

 

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.

Advertisement

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.

Advertisement

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.

Advertisement
Popular Mobile Brands
  1. Oppo Reno 14 5G Series Teased to Launch in India Soon
  2. Samsung Galaxy M36 5G India Launch Date and Key Features Revealed
  3. Nothing Phone 3 to Get New Glyph Matrix Interface on the Rear Panel
  4. OnePlus Bullets Wireless Z3 With Up to 36 Hours Battery Launched in India
  5. Google Messages Now Lets You Delete Messages for Everyone in RCS Chats
  6. Honor Magic V5 Confirmed to Launch on This Date
  7. World's Oldest Tailored Dress Found in Egyptian Tomb Dates Back Over 5,000 Years
  1. Fast Radio Bursts Reveal Universe’s Missing Matter Hidden in Cosmic Intergalactic Fog
  2. Apollo Astronauts Found Orange Glass Beads on the Moon, Scientists Now Know Why
  3. World’s Oldest Tailored Dress Found in Egyptian Tomb Dates Back Over 5,000 Years
  4. Ancient Footprints in White Sands Confirm Humans Reached America 23,000 Years Ago
  5. Humanoid Robot Achieves Controlled Flight Using Jet Propulsion and AI Systems
  6. Curiosity Rover Reaches Uyuni Quad, Begins New Mars Mapping and Surface Analysis Campaign
  7. NASA to Gather Reentry Imagery of European Commercial Capsule Using High-Altitude Aircraft
  8. ESA's Proba-3 Unveils First-Ever Artificial Solar Eclipse Images from Precision Satellite Formation
  9. My Hero Academia Final Season OTT Release Date Revealed: Everything You Need to Know
  10. NASA Study Reveals Correlation Between Earth’s Magnetic Field and Atmospheric Oxygen
Gadgets 360 is available in
Download Our Apps
Available in Hindi
© Copyright Red Pixels Ventures Limited 2025. All rights reserved.