Hugging Face Blog·· 19 小时前精选AI 评分66
Hugging Face 接入 TorchTPU,Transformers 与 Diffusers 原生支持 TPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 宣布 TorchTPU 作为原生 PyTorch 后端接入其生态,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。
推荐理由
原文给出 TorchTPU 的安装命令与 Transformers、Diffusers、Kernels 的接入方式,可据此判断 PyTorch 用户迁移到 TPU 的成本变化。
来源:Hugging Face Blog · huggingface.co