RecodeX 重构消息,Google与Meta联合发布开源项目TorchTPU,为Google TPU提供原生PyTorch后端支持。开发者迁移模型时,只需将代码中的 .to("cuda") 改为 .to("tpu"),即可在TPU上运行PyTorch模型。该项目旨在降低PyTorch生态向TPU迁移的适配成本。
Google与Meta开源TorchTPU,为TPU提供PyTorch原生后端
AI 辅助判断积极AI 看多仅供信息参考,不构成投资建议
原始信息来源Aihotaihot.news
查看原文 ↗