RecodeX 重构消息,Google与Meta联合发布开源项目TorchTPU,为Google TPU提供原生PyTorch后端支持。开发者迁移模型时,只需将代码中的 .to("cuda") 改为 .to("tpu"),即可在TPU上运行PyTorch模型。该项目旨在降低PyTorch生态向TPU迁移的适配成本。