Hugging Face 博客：TorchTPU 让 PyTorch 原生支持 TPU

Hugging Face 介绍原生 PyTorch 后端 TorchTPU，TPU 成为 PyTorch 的普通设备，Transformers、Diffusers 和 Kernels 可直接在 TPU 上运行，加载模型时传 device_map="tpu" 即可。