MLX — MLX 0.24.1 文档

MLX 是由 Apple 机器学习研究推出的类似 NumPy 的数组框架,专为 Apple 芯片上的高效灵活机器学习设计。

  • Python API 与 NumPy 相近但有少数例外,且有功能完备的 C++ API 且遵循 Python API。
  • 与 NumPy 的主要区别:可组合函数转换用于自动微分等;计算是惰性的,数组按需实例化;多设备,操作可在多种支持设备上运行。
  • 设计受[PyTorch]、[Jax]、[ArrayFire]等框架启发,有统一内存模型,数组在共享内存中,操作可在不同设备类型上进行无需数据拷贝,目前支持 CPU 和 GPU。
  • 用法包括:[Quick Start Guide]等多个方面的使用指南。
  • Python API 参考:包含数组、数据类型等多个方面的参考文档。
阅读 14
0 条评论