PyTorch Blog· The TorchTPU Team†·· 3 小时前精选AI 评分75
Google 与 Meta 联合推出 TorchTPU，让 PyTorch 原生运行在 TPU 上
Introducing TorchTPU
AI 导读
Google 与 Meta 合作开源 TorchTPU，为 Google TPU 提供原生 PyTorch 后端，把 model.to("cuda") 改成 model.to("tpu") 即可运行现有工作负载。
推荐理由
原文给出 TorchTPU 的架构取舍、精度模式与多组实测数据，可据此判断 PyTorch 工作负载迁移到 TPU 的实际代价。
来源：PyTorch Blog · pytorch.org