跳到正文
Hacker News 热帖· ilreb·· 3 小时前精选AI 评分78

Google 开源 TorchTPU:原生支持 TPUs 的 PyTorch 后端

Google open-sources TorchTPU, a native PyTorch back end for TPUs

AI 导读

Google 开源了 TorchTPU,这是一个为 Google Tensor Processing Units (TPUs) 提供原生支持的 PyTorch 后端项目。

推荐理由

原文提供了 TorchTPU 的开源仓库、安装指南及分布式训练示例,开发者可据此评估在 TPU 上运行 PyTorch 模型的可行性。

来源:Hacker News 热帖 · github.com