Ouvrir le menu principal

MacGeneration

Recherche

MLX : un framework de machine learning adapté aux puces Apple

Anthony Nelzin-Santos

jeudi 07 décembre 2023 à 17:00 • 24

Logiciels

Si vous vous êtes déjà intéressé au machine learning, vous avez certainement croisé la bibliothèque NumPy de manipulation des tableaux et matrices, une brique fondamentale de l’édifice scientifique du langage de programmation Python. Apple présente MLX, un nouveau framework suivant NumPy « de très près », à ceci près qu’il prend en compte les particularités de l’architecture Apple Silicon pour augmenter sensiblement ses performances.

Image Apple.

Conçu « par des chercheurs en machine learning pour les chercheurs en machine learning », MLX marche dans les pas de JAX, le framework de manipulation de fonctions numériques créé par Google pour suivre la structure et les habitudes de NumPy. Les ingénieurs de la firme de Cupertino se sont aussi inspiré de la bibliothèque de machine learning PyTorch, originellement développée par Meta AI et utilisée par la plupart des systèmes de deep learning, ainsi que du système de calcul parallélisé ArrayFire.

MLX possède l’énorme avantage d’être capable de réaliser des opérations sur le processeur et le circuit graphique sans dupliquer les données, en tirant pleinement parti de la mémoire unifiée. À l’exception d’une poignée de transformations et de l’évaluation paresseuse, qui permet d’éviter de réaliser un calcul dont les résultats ne seraient finalement pas nécessaires, il sera immédiatement familier aux utilisateurs de NumPy.

Autrement dit, MLX permettra de tirer toute la substantifique moelle des puces Apple sans pour autant réinventer la roue. Apple propose une API en Python, très proche de NumPy et donc adressée à la communauté des chercheurs, mais aussi une API en C++, qu’elle utilise probablement elle-même. MLX est distribué sous licence MIT avec quelques exemples d’utilisation : l’entrainement d’un grand modèle de langage, la génération de texte avec LLaMA et d’images avec Stable Diffusion, ou encore la reconnaissance de la voix avec Whisper.

iOS 18, Siri, iWork : Apple se prépare à ajouter de l

iOS 18, Siri, iWork : Apple se prépare à ajouter de l'IA dans tout son écosystème

Apple expérimente une nouvelle technologie de génération d

Apple expérimente une nouvelle technologie de génération d'images et de vidéos

Rejoignez le Club iGen

Soutenez le travail d'une rédaction indépendante.

Rejoignez la plus grande communauté Apple francophone !

S'abonner

Pour le producteur de La Maison, « Apple est la pire boîte marketing de l’univers »

00:08

• 23


Upscaling vidéo et restauration photo plus rapides sur Mac/PC : VideoProc AI mis à jour est à - 62 % 📍

25/04/2025 à 17:44

• 0


Derniers jours Apple week à la Fnac : -10 % sur des MacBook Pro, iPhone 15 Pro à 16 Pro Max et iPad Pro M4

25/04/2025 à 16:00

• 6


dav1d, le décodeur AV1 le plus optimisé qu'Apple refuse d'utiliser

25/04/2025 à 14:00

• 3


Microsoft a conçu une publicité par IA et personne n’a rien vu

25/04/2025 à 13:45

• 22


Razer lance sa première souris ergonomique verticale face à Logitech

25/04/2025 à 12:15

• 9


Yahoo! veut acheter Chrome

25/04/2025 à 10:45

• 32


Slate veut repartir de zéro avec un pick-up électrique compact et épuré pour moins de 20 000 $

25/04/2025 à 10:00

• 100


iPadOS 19 verrait apparaître une barre de menus sur iPad

25/04/2025 à 09:24

• 47


Huit ans après DeX, Apple s’ouvrirait peu à peu à un bureau sur écran externe pour iOS 19

25/04/2025 à 09:23

• 48


John Giannandrea aurait aussi perdu l'équipe en charge de la robotique chez Apple

25/04/2025 à 07:39

• 10


Les procès antitrust contre les GAFAM se poursuivent aux USA, malgré les changements de présidence

24/04/2025 à 21:30

• 27


Prise en main de Supercharge, l’app à tout faire qui rend bien des services sur le Mac

24/04/2025 à 20:30

• 17


TSMC annonce la gravure en 1.4 nm, quand la Russie espère atteindre les 28 nm en 2030

24/04/2025 à 20:20

• 55


Données personnelles : comment Incogni supprime ce que les autres ne peuvent pas 📍

24/04/2025 à 18:17

• 0


Amazon va fêter ses 25 ans en France avec une semaine de promos du 30 avril au 6 mai

24/04/2025 à 17:30

• 18