更多博客
博客必读进阶

Hugging Face ❤️ TorchTPU:原生支持 TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

博客文章,介绍 TorchTPU 这一原生 PyTorch 后端,使 TPU 成为 PyTorch 中的普通设备;文中称 Transformers、Diffusers 和 Kernels 可在 TPU 上原生运行,从 Hub 选择模型后发送到 "tpu" 即可。文章还简介了 TPU 作为 Google 自研 AI 加速器的情况与可用规模,并对比了此前需经 torch_xla 的旧方式及其限制。面向希望在 TPU 上运行 PyTorch 模型的用户。

收录时间
2026-10-26