MLX：苹果公司在AI赛道上的新动作，为硅芯片量身定做

苹果公司最近发布的MLX机器学习框架，标志着该公司在人工智能和机器学习技术方面迈出了重要一步。MLX专为苹果硅芯片设计，是一个公开可用的框架，旨在简化特别是在苹果自家硬件上的机器学习开发过程。

MLX因其针对苹果硅芯片的优化而脱颖而出，这表明苹果公司继续专注于更紧密地整合其硬件和软件能力。该框架建立在苹果基于ARM架构的硅芯片上，无需仿真或转换即可高效运行。这确保了MLX可以充分利用苹果的神经网络引擎、GPU和内存子系统的全部功率，以提供快速高效的机器学习计算​​。

https://github.com/ml-explore/mlx

该框架不仅功能强大，而且对开发人员来说非常熟悉，支持流行的机器学习库和语言，如Python、R和Julia。它包括一个与NumPy密切对齐的Python API，还提供了一个完整的C++ API。像mlx.nn和mlx.optimizers这样的高级包基于PyTorch，即使对于更复杂的模型也可以直接使用。MLX的数组框架允许懒计算、动态图形，以及将机器学习工作分布在CPU和GPU上的潜力。MLX的一个关键特性是其数组存储在共享内存中，使得所有设备类型上的操作无需移动数据即可执行​​。

在应用方面，MLX非常灵活。它支持一系列机器学习任务，包括训练Transformer模型、使用LoRA等工具进行微调，以及与现有AI开源项目（如Stable Diffusion、Whisper或LLama）集成。这种灵活性使其成为研究人员和开发人员在各种AI和机器学习项目上的宝贵工具​​。

通过发布MLX，苹果公司展示了其对机器学习的承诺，无论是在硬件还是软件方面。MLX与苹果现有框架和工具的整合，有望增强苹果生态系统中机器学习应用的功能和可用性。该公司对其设备上AI应用的愿景正在逐渐展开，MLX是朝着这个方向迈出的重要一步​​​​。

作者声明本文无利益相关，欢迎值友理性交流，和谐讨论～