Hugging Face Blog·· 20 小时前精选AI 评分71
Hugging Face 接入 TorchTPU：Transformers、Diffusers 与 Kernels 原生支持 TPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 宣布 Transformers、Diffusers 和 Kernels 已原生支持 TorchTPU，用户把模型加载到 "tpu" 设备即可运行，不再需要 Optimum TPU。
推荐理由
Hugging Face 生态原生接入 TPU 后，PyTorch 用户不再需要 torch_xla 或 JAX 就能把模型放到 TPU 上运行。
来源：Hugging Face Blog · huggingface.co