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 生成图像的示例。