Repolode
ml-explore

ml-explore/mlx-swift

MLX 的 Swift 接口,让开发者在 Apple 芯片上用 Swift 进行机器学习研究与实验。

Star
2,066
Fork
371
未关闭 Issue
56
最近提交
21小时前

这是什么?

MLX Swift 是 MLX 的 Swift 接口。MLX 是面向 Apple 芯片的机器学习数组框架,该项目把它的数组、神经网络和优化器等模块带到 Swift 语言中,方便在 Apple 设备上做研究和实验。项目以 Swift 包的形式发布,包含 MLX、MLXNN、MLXOptimizers、MLXRandom 等库。

可以通过 Xcode 或 SwiftPM 引入,使用 xcodebuild 构建以编译 Metal 着色器。项目也提供 CMake 构建方式,支持在 Linux 上使用 CPU 后端,并可选 CUDA GPU 后端。

大语言模型和视觉语言模型的实现位于独立的 mlx-swift-lm 项目中,mlx-swift-examples 仓库则包含 MNIST 训练、聊天应用、Stable Diffusion 等示例。

核心功能

  • 01为 Apple 芯片上的 MLX 数组框架提供 Swift 接口
  • 02以独立的 Swift 包产品提供 MLX、MLXNN、MLXOptimizers、MLXRandom 等库
  • 03可通过 Xcode 或 SwiftPM 引入,编译 Metal 着色器需使用 xcodebuild
  • 04提供 CMake 构建,支持 Linux 原生编译,包括 CPU 和 CUDA GPU 后端

适用场景

  • 在 Mac 或 iOS 设备上用 Swift 原生运行机器学习实验
  • 开发在 Apple 芯片上本地训练或运行小型神经网络的应用
  • 将 MLX Python 研究代码迁移到 Swift 项目中

快速上手

  1. 1

    在 Package.swift 中添加 mlx-swift 依赖

    .package(url: "https://github.com/ml-explore/mlx-swift", from: "0.10.0")
  2. 2

    将 MLX、MLXNN 和 MLXOptimizers 库添加为目标依赖

  3. 3

    克隆仓库并拉取子模块后再本地构建

    git submodule update --init --recursive
  4. 4

    使用 Xcode 命令行工具构建并运行测试

    xcodebuild test -scheme mlx-swift-Package -destination 'platform=OS X'

使用前须知

  • 主要面向 Apple silicon,Metal GPU 后端以 macOS 为主。
  • 命令行 SwiftPM 无法编译 Metal 着色器,完整构建需要使用 Xcode。
  • Linux 默认使用 CPU 后端,CUDA 支持仅限 Linux。

标签

常见问题

mlx-swift 是什么?
MLX 的 Swift 接口,让开发者在 Apple 芯片上用 Swift 进行机器学习研究与实验。 MLX Swift 是 MLX 的 Swift 接口。MLX 是面向 Apple 芯片的机器学习数组框架,该项目把它的数组、神经网络和优化器等模块带到 Swift 语言中,方便在 Apple 设备上做研究和实验。项目以 Swift 包的形式发布,包含 MLX、MLXNN、MLXOptimizers、MLXRandom 等库。
mlx-swift 可以免费使用吗?用的什么许可证?
mlx-swift 以 MIT 许可证开源。用于商业产品前,请先确认许可证条款。
mlx-swift 还在维护吗?
活跃维护中。mlx-swift 最近一次提交是在 2026年10月19日。
mlx-swift 适合用来做什么?
在 Mac 或 iOS 设备上用 Swift 原生运行机器学习实验 开发在 Apple 芯片上本地训练或运行小型神经网络的应用 将 MLX Python 研究代码迁移到 Swift 项目中
mlx-swift 怎么安装和运行?
在 Package.swift 中添加 mlx-swift 依赖: .package(url: "https://github.com/ml-explore/mlx-swift", from: "0.10.0") 将 MLX、MLXNN 和 MLXOptimizers 库添加为目标依赖 克隆仓库并拉取子模块后再本地构建: git submodule update --init --recursive 使用 Xcode 命令行工具构建并运行测试: xcodebuild test -scheme mlx-swift-Package -destination 'platform=OS X'
在哪里获取 mlx-swift?
源代码和安装说明在 GitHub 上:https://github.com/ml-explore/mlx-swift

概要由 AI 根据项目 README 生成;GitHub 数据每日更新。

相关项目