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

Hugging Face 携手 TorchTPU 实现 PyTorch 原生 TPU 支持

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布 TorchTPU 成为 PyTorch 的原生 TPU 后端,Transformers、Diffusers 和 Kernels 现可在 TPU 上直接运行。

推荐理由

原文给出了从 torch_xla 到原生 TPU 的安装命令和代码写法,并附 Qwen3 与 Gemma 4 的实测加速数据,读者可据此迁移和评估 TPU 方案。

来源:Hugging Face Blog · huggingface.co