mlx

MLX是一个针对Apple芯片设计的机器学习数组框架。它提供了类似NumPy的Python API以及功能完整的C++、C和Swift接口，简化了复杂模型的构建过程。核心功能包括自动微分、自动向量化、计算图优化等可组合函数变换，支持延迟计算与动态图构建，允许在CPU或GPU上执行操作而无需数据迁移，这得益于其统一内存模型。该框架适用于需要高效训练和部署模型的研究场景，特别是那些寻求快速探索新想法并希望获得简洁设计体验的机器学习研究人员。