Hugging Face Blog·· 18 小时前精选AI 评分66
Hugging Face 推出原生 PyTorch TPU 后端 TorchTPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 推出原生 PyTorch 后端 TorchTPU，把 TPU 变成 PyTorch 的又一个设备，Transformers、Diffusers 和 Kernels 可在 TPU 上原生运行。
推荐理由
原文给出 TorchTPU 的原生后端定位、安装命令与 Transformers、Diffusers 的接入方式，读者可据此在 TPU 上直接跑 PyTorch 模型。
来源：Hugging Face Blog · huggingface.co