Search

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
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.

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.

 
Show Full Article
Please wait...
Advertisement
Popular Mobile Brands
  1. Lava Blaze Dragon 5G Launched in India With Snapdragon 4 Gen 2 SoC
  2. Honor Pad X7 With 8.7-Inch Display and Snapdragon 680 SoC Launched
  3. ChatGPT Agent Is Finally Available to OpenAI's Paid Subscribers
  4. This New Tracking System Can Use Just Wi-Fi Signals to Identify You
  1. Who-Fi: An AI-Powered Wi-Fi Technology That Can Identify and Track Individuals Without Cameras
  2. NASA’s X-59 Moves Closer to First Flight with Advanced Taxi Tests and Augmented Vision
  3. Unusual Plasma Waves Above Jupiter’s North Pole Can Possibly Be Explained
  4. NASA to Live Stream SpaceX Crew-11 Launch Docking, Know How to Watch Online
  5. Apple Expands App Store Age Rating System With More Granular Categories
  6. Amazon Kindle Colorsoft Kids With 7-Inch Display and a Kid-Friendly Cover Launched
  7. Dying Light: The Beast Delayed to September for 'Extra Polishing Work'
  8. Flipkart Freedom Sale 2025 Date Revealed; Plus, VIP Members To Get Early Access
  9. Huawei Mate XT 2 Colour Options Leaked; Could Be World’s Second Mass-Produced Tri-Fold Phone
  10. Blender Is Working on a Standalone App for iPad Pro With Apple Pencil Support
Gadgets 360 is available in
Download Our Apps
App Store App Store
Available in Hindi
App Store
© Copyright Red Pixels Ventures Limited 2025. All rights reserved.
Trending Products »
Latest Tech News »