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

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 capa...