跳到正文
Hugging Face 官方团队博客· Irakli Salia·· 14 小时前AI 评分70

PyTorch 2.15 的 MPS 后端大更新:FP8 上 Mac,MoE 解码提速 5-6.5 倍

MPS in PyTorch 2.15

AI 导读

Hugging Face 的 PyTorch 团队介绍 PyTorch 2.15 的 MPS 后端更新,将 FP8(torch.float8_e4m3fn)带到 Mac,并通过 Metal 内核加速语言模型解码。

来源:Hugging Face 官方团队博客 · huggingface.co