Python计算库

MLX

ml-explore.github.io

MLX是苹果机器学习研究团队推出的数组计算框架，核心定位是为苹果芯片平台的机器学习研发提供高效灵活的底层支持。框架API设计与NumPy高度相似，同时支持可组合函数转换、惰性计算、多设备统一内存访问三大核心特性，可大幅降低苹果生态下机器学习模型的开发与部署成本。目标用户覆盖机器学习研究者、算法工程师、苹果生态应用开发者等群体，适用于模型原型快速验证、苹果设备端模型性能优化、深度学习基础算子开发等多种场景。