As the prominence of artificial intelligence (AI) continues to grow across industries, Apple’s machine learning research team is making strides with the release of MLX—a machine learning framework tailored for efficient and flexible operations on Apple’s silicon chips.
Unveiling MLX, Apple aims to provide a powerful tool inspired by established frameworks like PyTorch, Jax, and ArrayFire. What sets MLX apart is its unified memory model, as highlighted by Apple on GitHub. The ML research team emphasizes that MLX’s array design operates in shared memory, allowing operations on MLX arrays to be executed on various supported devices, including the CPU and GPU, without necessitating data copies.
Awni Hannun from Apple’s ML team shared the exciting news on X, stating, “Just in time for the holidays, we are releasing some new software from Apple machine learning research.” Hannun described MLX as an efficient machine learning framework specifically crafted for Apple silicon, extending its functionality to users’ laptops.
MLX is designed as a NumPy-like array framework, ensuring efficiency and flexibility in machine learning tasks on Apple silicon. Apple’s ML team emphasizes a Python API that closely mirrors NumPy, with additional features. Furthermore, MLX boasts a comprehensive C++ API that aligns closely with the Python API.
Last month, Apple’s CEO, Tim Cook, underscored the company’s significant investment in AI during an earnings call. Cook emphasized the responsible integration of AI for ongoing product enhancements. AI and machine learning, according to Cook, are fundamental technologies integral to nearly every product shipped by the company.
Cook cited examples of AI-driven features in recent iOS releases, such as Personal Voice and Live Voicemail, and life-saving features like fall detection, crash detection, and ECG on the Apple Watch. These advancements, Cook asserted, wouldn’t be possible without AI.
Acknowledging Apple’s sustained efforts in AI, Cook stated, “I’m not going to get into details about what it is because we really don’t do that. But you can bet that we’re investing, we’re investing quite a bit, we’re going to do it responsibly, and you will see product advancements over time.”
In August, Cook had already hinted at Apple’s extensive work on generative AI and other models, emphasizing the tech giant’s commitment to pushing the boundaries of AI technologies. As MLX takes center stage, Apple reinforces its dedication to advancing AI capabilities and delivering innovative solutions to its users.