跳到正文
Hugging Face Blog·· 3 天前精选AI 评分75

Hugging Face 宣布通过 TorchTPU 原生支持 Google TPU

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布旗下 Transformers、Diffusers 和 Kernels 库通过原生 PyTorch 后端 TorchTPU 获得 TPU 原生支持,不再依赖 Optimum TPU 或 torch_xla。

推荐理由

此前在 TPU 上运行 PyTorch 往往受限于图重编译或需改写 JAX,原生后端接入后开发者能直接将 TPU 视作普通设备调用。

来源:Hugging Face Blog · huggingface.co