跳到正文
原文
国际大厂·· 3 小时前精选AI 评估 · 77/100分数表示内容值得关注的程度,依据实质份量、信息增量、证据强度、读者相关性和可用性,按内容类型加权。显示分数为两次独立评分的平均值。AI 判断可能有误,评分不代表事实已被核实,也不同于热点榜的讨论热度。

Hugging Face 支持 TorchTPU,Transformers 与 Diffusers 可原生跑在 TPU 上

Hugging Face ❤️ TorchTPU: TPUs, Natively

AI 导读

Hugging Face 宣布接入原生 PyTorch 后端 TorchTPU,Transformers、Diffusers 和 Kernels 现在可直接在 TPU 上运行,不再需要 torch_xla 或 Optimum TPU。

推荐理由

原文给出了安装命令、各库的接入方式和实测吞吐数据,读者可据此判断现有 PyTorch 工作流迁移到 TPU 的成本。

来源:国际大厂 · huggingface.co