Google 与 Meta 发布 TorchTPU：PyTorch 原生 TPU 后端开源
Introducing TorchTPU
阅读原文
本站未展示全文，请前往来源网站阅读。
AI 导读
Google 与 Meta 联合发布开源项目 TorchTPU，为 Google TPU 提供原生 PyTorch 后端，模型迁移只需把 .to("cuda") 改为 .to("tpu")。
推荐理由
原文给出 TorchTPU 的架构设计、模型覆盖和性能数据，读者可以据此评估 PyTorch 工作流迁移 TPU 的可行性。
来源：PyTorch Blog · pytorch.org