跳到正文
Hugging Face Blog·· 19 小时前精选AI 评分76

Hugging Face 发布 TorchTPU 原生 PyTorch 后端

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 推出 TorchTPU 原生 PyTorch 后端,让模型、Diffusers 流水线和 Kernels 可直接在 TPU 上运行。

推荐理由

把 TPU 作为 PyTorch 原生后端后,Transformers 与 Diffusers 无需专用分支即可沿用既有代码和部署流程。

来源:Hugging Face Blog · huggingface.co