MLX
Plus d'infos
- Créé par
- Apple
- Année
- Propriétaire
- Apple
- Type
- Framework ML pour Apple Silicon
- Licence
- MIT
- Site officiel
- ml-explore.github.io
- Code source
- github.com/ml-explore/mlx
Le framework ML open source d'Apple, optimisé pour l'Apple Silicon — il exploite la mémoire unifiée (le CPU et le GPU partagent la RAM) pour éviter les copies de données.
Sorti en décembre 2023. L'API rappelle volontairement PyTorch, avec une syntaxe façon NumPy ; l'évaluation paresseuse permet de fusionner les opérations avant leur exécution sur le GPU. Quantification intégrée, support de Mistral, Llama, Phi, Stable Diffusion.
Son terrain de prédilection : l'inférence locale et le fine-tuning à petite échelle sur la série M. Un M2 Max avec 64 Go de mémoire unifiée peut faire tourner un Llama 3.1 70B en Q4 localement — ce qui demanderait sinon un A100. Ce n'est pas un concurrent des grandes stacks d'entraînement (JAX/PyTorch), mais un sérieux candidat pour l'« IA grand public ».