Apple Launches Open-Source Tools for AI Developers Using Macs

MLX allows for Stable Diffusion and OpenAI’s Whisper to run on Apple hardware.

Ben Wodecki, AI Business

December 7, 2023

2 Min Read
AI concept image

This article was originally published on AI Business.

Apple has thrown its hat into the open-source AI ring by releasing a new range of open-source AI tools.

The MLX framework for machine learning is developed specifically for Apple’s Silicon processor architecture. Available via GitHub, MLX is designed to streamline machine learning model training and deployment on Apple hardware.

Apple said the design of MLX is “inspired” by other popular frameworks, including PyTorch, Jax, and ArrayFire. Where MLX differs, however, is the unified memory model - Arrays in MLX live in shared memory, while operations can be performed on any of the supported device types without performing data copies.

“The framework is intended to be user-friendly, but still efficient to train and deploy models. The design of the framework itself is also conceptually simple. We intend to make it easy for researchers to extend and improve MLX with the goal of quickly exploring new ideas,” the MLX repository reads.

MLX has a Python API that closely follows NumPy – a popular library for Python programming. MLX also contains a C++ API and higher-level packages that follow PyTorch for building more complex models.

Other key features of MLX include:

  • Composable function transformations: MLX has composable function transformations for automatic differentiation, automatic vectorization and computation graph optimization.

  • Lazy computation: Computations in MLX are lazy, with arrays only materializing when needed.

  • Dynamic graph construction: Computation graphs in MLX are built dynamically. If a user changes the shape of a function, arguments don’t trigger slow compilations – making debugging easier and intuitive.

Related:What Is Gemini? Google Launches New AI Software

Showcasing MLX’s abilities, Apple machine learning Research Scientist Awni Hannun posted a video on X (Twitter) that showed the seven billion parameter version of Meta’s LLaMA running on an M2 Ultra chip, found in both Mac Studio and Mac Pro, Apple’s high-end computing systems.

Other examples Apple showed off MLX include generating images with Stable Diffusion, speech recognition with OpenAI’s Whisper and parameter-efficient fine-tuning with LoRA.

Like most companies, Apple has been stepping up its AI work. Reports emerged in the summer that it’s working on its own web application-based chatbot service called ‘Apple GPT.’

Can MLX Be Used in Commercial Software?

Apple’s MLX is available under an MIT license which allows for broad freedom of use, including for commercial purposes.

The conditions of the license require copyright notice and permission notices to be included in all copies of the software.

Read more about:

AI Business

About the Author(s)

Ben Wodecki

Assistant Editor, AI Business

Ben Wodecki is assistant editor at AI Business, a publication dedicated to the latest trends in artificial intelligence.

AI Business

AI Business, an ITPro Today sister site, is the leading content portal for artificial intelligence and its real-world applications. With its exclusive access to the global c-suite and the trendsetters of the technology world, it brings readers up-to-the-minute insights into how AI technologies are transforming the global economy - and societies - today.

Sign up for the ITPro Today newsletter
Stay on top of the IT universe with commentary, news analysis, how-to's, and tips delivered to your inbox daily.

You May Also Like