Technology

Apple Dives into AI Development with New Frameworks and Libraries

Published December 6, 2023

Perceived as cautious in the AI domain, Apple has recently made a stealthy yet significant move by releasing a set of frameworks and model libraries. These tools are specially designed to harness the power of Apple's proprietary chips, potentially paving the way for generative AI applications on MacBooks.

The Emergence of MLX and MLX Data

Apple's machine learning team introduced MLX, a new machine learning framework tailored for developers to create models that maximize the efficiency of Apple Silicon. Alongside MLX, they have also unveiled MLX Data, a deep learning model library. Developers can access these tools via open-source platforms such as GitHub and PyPI.

Inspiration Behind MLX

The foundation of MLX was influenced by existing frameworks like PyTorch, Jax, and ArrayFire. A distinctive feature of MLX is its shared memory system; it allows tasks to run on supported devices, including CPUs and GPUs without the need for data relocation. It is designed to be developer-friendly while possessing the ability to train sophisticated AI models.

Flexibility of MLX Data

Apple researcher Awni Hannun highlighted MLX Data for its compatibility and efficiency. It is described as a 'framework agnostic' solution for data loading that facilitates integration with MLX, PyTorch, or Jax frameworks.

Apple's foray into AI isn't new, having integrated machine learning into its products for some time. However, these efforts have primarily been in machine learning, not in the rapidly growing sector of generative AI that rivals like Microsoft and Google are developing.

Apple's AI Strategy

In recent efforts, Apple began exploring foundational models that could be incorporated across its various services, marking a strategic move towards a broader application of AI.

Apple, AI, MLX