Hugging Face 联合 TorchTPU 为 Transformers、Diffusers、Kernels 带来 TPU 原生支持

Hugging Face 宣布通过 PyTorch 原生后端 TorchTPU，Transformers、Diffusers 和 Kernels 无需 torch_xla 或 JAX 即可原生运行在 TPU 上，模型只需 device_map="tpu" 即可加载。