写在前面:

本文章是关于 Apple 公司开源人工智能框架 MLX 的部分学习笔记。碍于本人学识有限,部分叙述难免存在纰漏,请读者注意甄别。

参考资料:


零、前言

MLX 是苹果公司于 2023 年 12 月发布的、专为 Apple Silicon 芯片(Arm 架构)设计的开源机器学习框架,旨在支持在 Apple Silicon 芯片上高效地训练和部署模型。

但是针对苹果此次发布的 MLX,呈现了两种截然不同的声音。

从头造轮子 or 重复造轮子?

  • 正方观点:MLX 作为一个针对苹果芯片优化的 PyTorch-style 神经网络架构,可能是苹果在 AI 开源领域,迄今为止最重要的举措。
  • 反方观点:苹果本可以跟 PyTorch 合作,发布针对苹果芯片优化的 PyTorch 后端。而如今开发者不得不重复工作,同时支持 MLXPyTorch