MLX | Apple 开源数组运算框架

本周 Apple 发起了它们在 AI 领域的攻势。Apple 机器学习团队开源了名为 MLX 的数组运算框架，面向机器学习领域设计，且专为 Apple Silicon 芯片进行优化。MLX 借鉴了 Numpy 和 PyTorch 等框架的 API 设计，提供了统一的 Python 和 C++ 接口，并且引入惰性计算等机制进一步优化性能。它也支持通过组合函数转换，自动实现微分和图优化等高级功能。