MLX — 苹果机器学习数组框架

事实

苹果为 Apple Silicon 打造的机器学习数组框架。

人话

MLX 让 Mac 的 CPU 和 GPU 共用一张案板：食材摆上去两边都能直接动手，不用端来端去。

借统一内存，可在苹果芯片训练、微调模型，也适合本地运行。

相关概念

• 统一内存
• 深度学习框架 PyTorch
• 本地大模型

Unified Memory
MLX 利用统一内存，让数组可在 CPU 与 GPU 间共享。

PyTorch
MLX 与 PyTorch 都用于张量计算和自动微分开发。

Local-LLM
MLX-LM 基于 MLX，让大模型可在 Apple Silicon 本地运行。