Apple dévoile des outils de développement d'IA open source pour les utilisateurs de Mac.

Apple a fait son entrée dans le paysage de l'IA open-source avec le lancement de sa suite innovante d'outils d'IA. Au cœur de cette initiative se trouve le cadre MLX, spécialement conçu pour l'architecture Silicon d'Apple. Accessible sur GitHub, MLX vise à optimiser l'entraînement et le déploiement de modèles d'apprentissage automatique sur le matériel Apple.

Apple décrit les principes de conception de MLX comme « inspirés » par des frameworks populaires tels que PyTorch, Jax et ArrayFire. Cependant, MLX se distingue par son modèle de mémoire unifiée : les tableaux résident dans une mémoire partagée, ce qui permet des opérations sur tous les types de dispositifs supportés sans nécessité de copier les données. Cette fonctionnalité fluidifie les flux de travail et améliore les performances.

« Le cadre vise à être convivial tout en garantissant un entraînement et un déploiement efficaces des modèles. Sa conception fondamentale est simple, permettant aux chercheurs d'étendre et d'améliorer facilement MLX pour explorer rapidement de nouvelles idées », indique le dépôt MLX.

Fonctionnalités clés de MLX :

- API Python similaire à NumPy : Le cadre MLX inclut une API Python qui ressemble étroitement à NumPy, facilitant la transition pour les utilisateurs familiers avec cette bibliothèque populaire. De plus, il propose une API C++ et des packages de niveau supérieur inspirés de PyTorch, permettant aux développeurs de construire des modèles plus sophistiqués.

- Transformations de fonctions composables : MLX prend en charge la différentiation automatique, la vectorisation automatique et l'optimisation des graphes de calcul grâce à des transformations de fonctions composables, simplifiant ainsi la gestion des tâches computationnelles complexes.

- Calcul paresseux : Le cadre utilise le calcul paresseux, ce qui signifie que les tableaux ne sont matérialisés que lorsque cela est nécessaire. Cette fonctionnalité optimise l'utilisation des ressources et améliore l'efficacité.

- Construction dynamique de graphes : MLX construit des graphes de calcul dynamiquement. Les utilisateurs peuvent modifier les formes de fonctions sans subir de retards dus à de longues compilations, ce qui simplifie le processus de débogage et améliore l'expérience utilisateur globale.

Lors d'une démonstration des capacités de MLX, Awni Hannun, scientifique en recherche en apprentissage automatique chez Apple, a partagé une vidéo montrant la version à sept milliards de paramètres du LLaMA de Meta fonctionnant sur une puce M2 Ultra, qui équipe le Mac Studio et le Mac Pro, les systèmes informatiques de pointe d'Apple. D'autres applications notables de MLX incluent la génération d'images avec Stable Diffusion, la reconnaissance vocale via Whisper d'OpenAI, et le fine-tuning efficace des paramètres avec LoRA.

Alors qu'Apple intensifie son intérêt pour l'IA, des rapports de cet été indiquent que l'entreprise développe également son service de chatbot basé sur une application web, connu sous le nom de « Apple GPT ».

Licences et utilisation commerciale :

MLX est disponible sous la licence MIT, offrant une grande liberté d'utilisation, y compris pour des applications commerciales. Toutefois, les utilisateurs doivent respecter les exigences de la licence en incluant des avis de copyright et de permission dans toutes les copies du logiciel.

Cet engagement envers les principes open-source démontre non seulement l'investissement d'Apple dans l'avancement de la technologie IA, mais il permet également aux développeurs, chercheurs et entreprises d'innover dans le domaine de l'IA en utilisant ses outils robustes.