Apple hat mit der Einführung seiner innovativen Suite offener AI-Tools in die Welt der Open-Source-KI eingetreten. Im Zentrum dieser Veröffentlichung steht das MLX-Framework, das speziell für die Silicon-Architektur von Apple entwickelt wurde. MLX ist auf GitHub zugänglich und optimiert das Training und die Bereitstellung von Machine-Learning-Modellen auf Apple-Hardware.
Apple beschreibt die Designprinzipien von MLX als "inspiriert" von beliebten Frameworks wie PyTorch, Jax und ArrayFire. Dennoch hebt sich MLX durch sein einheitliches Speichermodell hervor: Arrays befinden sich im gemeinsamen Speicher, was Operationen über alle unterstützten Gerätetypen ohne Datenkopierung ermöglicht. Diese Funktion vereinfacht Arbeitsabläufe und steigert die Leistung.
„Das Framework soll benutzerfreundlich sein und gleichzeitig ein effizientes Modelltraining und -deployment gewährleisten. Sein grundlegendes Design ist unkompliziert und ermöglicht es Forschern, MLX schnell zu erweitern und zu verbessern, um neue Ideen rasch zu erkunden“, heißt es im MLX-Repository.
Hauptmerkmale von MLX
- Python-API ähnlich wie NumPy: Das MLX-Framework bietet eine Python-API, die NumPy stark ähnelt und einen nahtlosen Übergang für Nutzer ermöglicht, die mit dieser beliebten Bibliothek vertraut sind. Zudem bietet es eine C++-API und höherstufige Pakete, die nach PyTorch modelliert sind, wodurch Entwickler komplexere Modelle erstellen können.
- Komponierbare Funktionsumwandlungen: MLX unterstützt automatische Differenzierung, automatische Vektorisierung und die Optimierung von Berechnungsgraphen durch komponierbare Funktionsumwandlungen, was die Handhabung komplexer Berechnungsaufgaben vereinfacht.
- Lazy Computation: Das Framework verwendet Lazy Computation, was bedeutet, dass Arrays nur bei Bedarf materialisiert werden. Diese Funktion optimiert die Ressourcennutzung und erhöht die Effizienz.
- Dynamische Graphkonstruktion: MLX erstellt Berechnungsgraphen dynamisch. Nutzer können die Form von Funktionen ändern, ohne Verzögerungen durch langsame Kompilierungen hinnehmen zu müssen, was den Debugging-Prozess vereinfacht und die Nutzererfahrung verbessert.
In einer Demonstration der Fähigkeiten von MLX teilte der Apple-Machine-Learning-Forscher Awni Hannun ein Video, das die sieben Milliarden Parameter umfassende Version von Metas LLaMA auf einem M2 Ultra-Chip zeigt, der die Mac Studio- und Mac Pro-Systeme von Apple antreibt. Weitere bemerkenswerte Anwendungen von MLX sind die Generierung von Bildern mit Stable Diffusion, die Spracherkennung über OpenAIs Whisper und die Implementierung von parameter-effizientem Fine-Tuning mit LoRA.
Während Apple seinen Fokus auf KI verstärkt, berichten Quellen aus diesem Sommer, dass das Unternehmen auch einen eigenen webanwendungsbasierten Chatbot-Service, bekannt als „Apple GPT“, entwickelt.
Lizenzierung und kommerzielle Nutzung
MLX steht unter der MIT-Lizenz, die umfangreiche Freiheiten für die Nutzung, einschließlich kommerzieller Anwendungen, bietet. Nutzer müssen jedoch die Anforderungen der Lizenz einhalten, indem sie Urheberrechts- und Genehmigungsmitteilungen in allen Kopien der Software anbringen.
Dieses Engagement für Open-Source-Prinzipien zeigt nicht nur Apples Investition in die Weiterentwicklung der KI-Technologie, sondern befähigt auch Entwickler, Forscher und Unternehmen, in der KI-Landschaft mit den leistungsstarken Tools zu innovieren.