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

Hugging Face 支持 TorchTPU:Transformers 与 Diffusers 原生运行在 TPU 上

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布接入原生 PyTorch 后端 TorchTPU,Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。

推荐理由

Hugging Face 官方说明 TorchTPU 让 Transformers 与 Diffusers 原生跑在 TPU 上,并给出实测加速数据。

来源:Hugging Face Blog · huggingface.co