MLX 是由 Apple 机器学习研究推出的类似 NumPy 的数组框架,专为 Apple 芯片上的高效灵活机器学习设计。
- Python API 与 NumPy 相近但有少数例外,且有功能完备的 C++ API 且遵循 Python API。
- 与 NumPy 的主要区别:可组合函数转换用于自动微分等;计算是惰性的,数组按需实例化;多设备,操作可在多种支持设备上运行。
- 设计受[PyTorch]、[Jax]、[ArrayFire]等框架启发,有统一内存模型,数组在共享内存中,操作可在不同设备类型上进行无需数据拷贝,目前支持 CPU 和 GPU。
- 用法包括:[Quick Start Guide]等多个方面的使用指南。
- Python API 参考:包含数组、数据类型等多个方面的参考文档。
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。