Hugging Face Blog·· 17 小时前精选AI 评分68
Hugging Face 支持 TorchTPU，Transformers 与 Diffusers 可原生跑在 TPU 上
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 宣布支持 PyTorch 原生 TPU 后端 TorchTPU，Transformers、Diffusers 和 Kernels 现在可直接把模型加载到 TPU 上运行。
推荐理由
HF 生态原生接入 TPU 的路径被压缩成一行 device_map，读者可据此判断迁移成本与已有 XLA 方案的差异。
来源：Hugging Face Blog · huggingface.co