mlx-swift

MLX Swift

安装 | 文档 | 示例 MLX Swift 是一个用于 MLX 的 Swift API。 MLX 是 Apple 硅基 机器学习 的阵列框架。MLX Swift 将 MLX 扩展到 Swift 语言,从而简化研究和实验 基于 Apple 晶片。

语言模型

mlx-swift-lm 提供了 LLM 和 VLM 实现。

示例

MLX Swift 具有 许多 示例, 包括:

• MNISTTrainer: 一个在 GPU 上运行的示例 无论是 iOS 还是 macOS,都可以下载 MNIST 训练数据并训练一个 LeNet。

• MLXChatExample: 一个在 iOS 和 macOS 上运行的聊天应用示例,支持 LLM 和 VLM。

MLX Swift 示例仓库 包含这些示例的完整代码和文档,包括来自 MLX Python 的代码。

guidelines on porting models