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

Hugging Face 发布 TorchTPU 原生 PyTorch TPU 后端

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 发布 TorchTPU,将 TPU 作为 PyTorch 原生后端,并让 Transformers、Diffusers 与 Kernels 无需 fork 即可原生运行。

推荐理由

文章提供安装命令、迁移变化和吞吐对比,可用于评估 PyTorch 工作流原生接入 TPU 的成本与收益。

来源:Hugging Face Blog · huggingface.co