Google 开源 TorchTPU,PyTorch 可原生运行在 TPU 上
TorchTPU is now open source: Run native PyTorch on TPUs
阅读原文
本站未展示全文,请前往来源网站阅读。
AI 导读
Google Cloud 在 PyTorch Conference 2026 宣布 TorchTPU 作为开源项目发布在 GitHub,TPU 由此成为与 GPU、CPU 并列的 PyTorch 原生选项,由 Google 与 Meta PyTorch 团队共同开发,现有 PyTorch 工作负载只需少量代码修改即可在 TPU 上训练、推理和部署。
推荐理由
TorchTPU 开源后 TPU 成为 PyTorch 原生选项,熟悉的工作流可以少改代码直接迁移,值得评估训练和推理成本。
来源:Google Cloud: Databases · cloud.google.com