Google 与 Meta 发布 TorchTPU：为 TPU 提供原生 PyTorch 后端

OmniTools 10月21日消息，PyTorch 官方博客 10 月 21 日消息，Google 与 Meta 发布开源项目 TorchTPU，为 Google TPU 提供原生 PyTorch 后端。

据项目介绍，借助该后端，模型迁移到 TPU 时只需把代码中的 .to("cuda") 改为 .to("tpu")，无需进行其他改写。

目前该项目已以开源形式发布，具体性能表现与适配范围尚未在摘要中披露。