ml-explore/mlx-swift

解決的問題

MLX Swift 提供 MLX 數組框架的原生 Swift API，讓開發者能在 Apple silicon 上直接使用 Swift 語言進行機器學習研究與實驗。

運作方式

它作為 MLX 框架的 Swift 扩展，允許使用者使用 Swift 建立與訓練模型。支援 Xcode、SwiftPM 與 CMake 等多種建構系統，並提供 Metal（macOS/iOS）與 CUDA（Linux）後端。

適用對象

希望在 macOS、iOS 與 Linux 上利用 Apple silicon 性能進行機器學習任務的 Swift 開發者與機器學習研究人員。

主要特色

• MLX 數組框架的原生 Swift API。
• 透過配套的 mlx-swift-lm 套件支援 LLM 與 VLM 實作。
• 支援 macOS、iOS 與 Linux 的跨平台使用。
• 包含在 MNIST 上訓練 LeNet、使用 LLM/VLM 運行聊天應用，以及使用 Stable Diffusion 生成影像的範例。