Apple a publié le cadre d'apprentissage automatique MLX pour Apple Silicon.

Par: Bohdan Kaminskyi | 08.12.2023, 16:31
Apple a publié le cadre d'apprentissage automatique MLX pour Apple Silicon.
Zhiyue/Unsplash.

Apple a dévoilé des frameworks et des bibliothèques de modèles conçus pour fonctionner sur ses puces.

Voici ce que nous savons

Le groupe de recherche sur l'apprentissage automatique d'Apple a publié MLX, un cadre permettant d'exécuter efficacement des modèles sur les puces Silicon d'Apple. La société a également présenté la bibliothèque MLX Data, qui contient des modèles d'apprentissage profond pré-entraînés.

Les deux produits sont disponibles en libre accès sur GitHub et PyPI.

Selon les développeurs, lors de la création de MLX, ils se sont concentrés sur des frameworks populaires tels que PyTorch, Jax et ArrayFire. Contrairement à ces derniers, MLX exécute toutes les tâches sur les appareils pris en charge (actuellement les CPU et les GPU) sans déplacer les données.

Selon Computerworld, le cadre est facile à apprendre mais suffisamment puissant pour entraîner des modèles d'IA complexes tels que le lama de Meta et la diffusion stable.

Selon Awni Hannun d'Apple, MLX Data est un package indépendant du framework pour un chargement efficace des données. Il est compatible avec MLX, PyTorch ou Jax.

Les cadres présentés sont axés sur l'apprentissage automatique plutôt que sur les applications populaires d'IA générative. Cela dit, le géant technologique s'efforce d'éviter le mot "IA" dans ses présentations.

Source : The Verge The Verge