MLX-vs-Pytorch使用交流:一个比较Apple Silicon GPU上PyTorch和MLX性能的基准测试项目,通过5个不同场景的测试(包括Transformer语言模型训练、BERT微调、Whisper推理等),全面对比两个框架在M1/M3系列芯片上的性能表现,为苹果设备上的AI项目选型提供参考 MLX-vs-Pytorch的特点: 1. 对比PyTorch和MLX在Apple Silicon GPU上...
苹果机器学习团队的Awni Hannun在X-note中将该软件称为:「……专为苹果芯片设计的高效机器学习框架」.MLX使用起来类似PyTorch、Jax和ArrayFire等现有框架.但是MLX针对Apple芯片进行了优化同时MLX增加了对统一内存模型的支持,这也意味着阵列位于共享内存中,并且可以在任何支持的设备类型上执行操作,而无需执行数据复制.那么...