Hacker News· gmays·· 1 小时前精选AI 评分80
Google 与 Meta 合作推出 TorchTPU，让 PyTorch 原生运行在 TPU 上
Introducing TorchTPU
AI 导读
Google 与 Meta 的 PyTorch 团队合作推出开源项目 TorchTPU，为 Google TPU 提供原生 PyTorch 后端，模型迁移只需把 .to("cuda") 改成 .to("tpu")。
推荐理由
Google 与 Meta 合作推出原生 PyTorch TPU 后端，文中给出迁移方式、性能数据与已验证模型范围，可据此判断 TPU 上跑 PyTorch 的落地成本。
来源：Hacker News · pytorch.org