Hugging Face Blog·· 23 小时前精选AI 评分81
Hugging Face 推出 TorchTPU,让 PyTorch 原生支持 TPU
Hugging Face ❤️ TorchTPU: TPUs, Natively
AI 导读
Hugging Face 推出 TorchTPU,为 PyTorch 增加原生 TPU 后端,Transformers、Diffusers 和 Kernels 可在 TPU 上运行。用户可通过 `device_map="tpu"` 加载模型,也可使用 `torch.compile` 和分布式训练;在已运行的 TPU 测试中,96.5% 通过,报告的 34 个问题已有 19 个修复。
推荐理由
TorchTPU让Transformers、Diffusers和Kernels可原生运行于TPU,展示了PyTorch用户接入Google加速器的路径变化。
来源:Hugging Face Blog · huggingface.co