Apple выпустила фреймворк машинного обучения MLX для Apple Silicon

Zhiyue/Unsplash

Компания Apple представила фреймворки и библиотеки моделей, предназначенные для работы на ее чипах. 

Что известно

Группа Apple по исследованиям в области машинного обучения выпустила MLX — фреймворк для эффективной работы моделей на чипах Apple Silicon. Также компания представила библиотека MLX Data с предобученными моделями глубокого обучения. 

Оба продукта доступны в открытом доступе на ресурсах GitHub и PyPI.

По словам разработчиков, при создании MLX они ориентировались на такие популярные фреймворки, как PyTorch, Jax и ArrayFire. Однако в отличие от них, MLX выполняет все задачи на поддерживаемых устройствах (в настоящее время CPU и GPU) без перемещения данных. 

Как сообщает Computerworld, фреймворк прост в освоении, но достаточно мощный для обучения сложных моделей ИИ вроде Meta's Llama и Stable Diffusion.

По словам Ауни Ханнуна (Awni Hannun) из Apple, MLX Data — независимый от фреймворка пакет для эффективной загрузки данных. Он совместим с MLX, PyTorch или Jax. 

Представленные фреймворки ориентированы на машинное обучение, а не на популярные приложения генеративного ИИ. При этом техгигант старается избегать самого слова "ИИ" в своих ключевых презентациях.

Источник: The Verge