跳到正文
Hugging Face Blog·· 16 小时前AI 评分50

PyTorch 2.15 中 MPS 后端性能大幅提升

MPS in PyTorch 2.15

AI 导读

PyTorch 2.15 的 MPS 后端在 Mac 上实现了多项性能优化,使 Qwen3.6-35B-A3B 等 MoE 模型的解码速度比 2.14 版快 5–6.5 倍。视频生成内存占用显著降低,Wan2.2 的 VAE 解码峰值内存从 270 GiB 降至约 8 GiB。该版本还引入了 FP8 张量支持以节省显存,并提升了线性层、softmax 等操作的执行效率。

来源:Hugging Face Blog · huggingface.co