跳到正文
Hugging Face Blog·· 17 小时前精选AI 评分83

Hugging Face 推出 TorchTPU,为 PyTorch 原生支持 TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 推出 TorchTPU,为 PyTorch 增加原生 TPU 后端,使 Transformers、Diffusers 和 Kernels 可在 TPU 上运行。用户可用常规 PyTorch 代码加载模型,并使用 torch.compile、分布式训练和 TPU kernels;测试中已运行的 TPU 测试有 96.5% 通过,34 个问题中 19 个已修复。

推荐理由

文章说明 TorchTPU 如何把 TPU 接入 PyTorch 及 Hugging Face 生态,并提供性能数据与测试通过率,便于了解其支持范围和当前成熟度。

来源:Hugging Face Blog · huggingface.co