跳到正文
千机 API
原文
Hugging Face Blog·· 23 小时前精选AI 评分81

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

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

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

推荐理由

TorchTPU让Transformers、Diffusers和Kernels可原生运行于TPU,展示了PyTorch用户接入Google加速器的路径变化。

来源:Hugging Face Blog · huggingface.co