【MLX:苹果芯片的深度计算框架,旨在为机器学习研究人员提供简单易用、高效训练和部署模型的工具,具有与NumPy相似的Python API和与之相似的C++ API。MLX支持自动微分、自动向量化和计算图优化等可组合的函数变换,具有延迟计算和动态图构建的特性,可在CPU和GPU上运行。与其他框架不同,MLX采用统一内存模型,数组存储在共享内存中,可以在不移动数据的情况下在不同设备上执行操作。MLX的设计受到NumPy、PyTorch、Jax和ArrayFire等框架的启发】’MLX - MLX: An array framework for Apple silicon' ml-explore GitHub: github.com/ml-explore/mlx #开源# #机器学习#
![]() |
创作新主题 |