第 2026-10-20 期 · 行业追踪 · 开源 · 芯片

Hugging Face支持TorchTPU原生PyTorch后端

Hugging Face支持TorchTPU原生PyTorch后端

Hugging Face宣布TorchTPU为PyTorch带来原生TPU后端,Transformers、Diffusers和Kernels可直接在TPU上运行,只需将模型发送到“tpu”。该后端包含适配TPU的Pallas内核和分布式训练支持,此前所需的Optimum TPU不再必需。

Hugging Face Blog2 天前
查看原文 ↗