ml-explore/mlx-swift-examples

Examples using MLX Swift

解决的问题

本仓库为使用 MLX Swift 的开发者提供了一系列实用示例和可复用的库,帮助他们了解如何在 Apple Silicon 上实现 AI 模型和数值计算任务。它弥合了原始 MLX Swift 框架与在 iOS 和 macOS 上构建实际应用程序之间的差距。

如何工作

该项目将示例分为三大类:AI 应用(如聊天应用和图像生成器)、用于快速测试的命令行工具,以及数值计算模拟。它利用 mlx-swift-lm 库实现语言模型,并提供了 Stable Diffusion 和 MNIST 的自定义实现。

适用人群

希望使用 MLX 框架在 Apple 平台(iOS 和 macOS)上构建 AI 驱动应用的 Swift 开发者。

主要亮点

  • 多样化的 AI 示例:包含 LLM、VLM 以及用于图像生成的 Stable Diffusion 的实现。
  • 跨平台支持:许多示例可在 iOS 和 macOS 上运行。
  • 微调支持:包含在 macOS 上对 LLM 进行 LoRA 微调的示例。
  • 数值计算:使用自定义 Metal 内核演示热扩散和曼德勃罗集渲染等通用数值任务。
  • 即用型工具:提供 CLI 工具,可快速从 Hugging Face 测试文本和图像生成模型。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目