Hugging Face Blog·· 18 小时前精选AI 评分78
Hugging Face 接入 TorchTPU，Transformers 与 Diffusers 可原生运行在 TPU 上
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 宣布接入原生 PyTorch 后端 TorchTPU，Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行，不再需要 torch_xla 或 Optimum TPU。
推荐理由
原文给出 TorchTPU 的安装命令、Transformers 与 Diffusers 的接入方式及实测加速数据，可据此评估 TPU 上跑 PyTorch 的迁移成本。
来源：Hugging Face Blog · huggingface.co