Hugging Face 联合 TorchTPU 为 Transformers、Diffusers、Kernels 带来 TPU 原生支持
Hugging Face 宣布通过 PyTorch 原生后端 TorchTPU,Transformers、Diffusers 和 Kernels 无需 torch_xla 或 JAX 即可原生运行在 TPU 上,模型只需 device_map="tpu" 即可加载。
推荐理由:原文给出从 torch_xla 迁移到原生 PyTorch 后端的具体用法和内核加速数据,读者可据此评估 TPU 工作流迁移成本。