广告合作
人工智能电脑

谷歌发布开源项目TorchTPU,PyTorch应用无需改写即可运行于TPU

谷歌称开发者只需修改少量配置,就能把现有PyTorch应用迁移到TPU和谷歌云,并同步推出TPU Agent Suite。

谷歌于2026年10月20日发布开源项目TorchTPU,使现有的PyTorch应用无需重写代码即可在TPU上运行。据digitaltoday.co.kr报道,开发者通过简单的配置修改,就能将应用迁移至TPU和谷歌云。

谷歌同时发布了「TPU Agent Suite」,用于协助应用迁移和修复程序错误,并更新了自家端侧AI运行时「LiteRT」,以改善与PyTorch应用的集成。

TorchTPU的兼容工作由谷歌与Meta、Hugging Face、Inferact(vLLM)以及RadixArk(SGLang)合作完成。

硬件适配的竞争

Meta于2016年开发了PyTorch,当时英伟达GPU是AI研究的标准硬件,因此PyTorch的许多功能是针对英伟达芯片设计的。有估算认为,85%的AI应用使用PyTorch。2022年,Meta将PyTorch移交给非营利组织Linux Foundation。

就在前一天,即2026年10月19日,亚马逊发布了自家的PyTorch版本TorchNeuron,可将PyTorch应用代码迁移到亚马逊的Trainium AI芯片和AWS上,同样无需重写代码。据TechCrunch报道,亚马逊正向PyTorch项目提交代码,以将多项功能与英伟达芯片解耦。

快速问答

TorchTPU是什么?

它是谷歌于2026年10月20日发布的开源项目,可让现有PyTorch应用无需重写代码即可在TPU上运行。

使用TorchTPU需要重写代码吗?

不需要。开发者通过简单的配置修改,就能将应用迁移至TPU和谷歌云。

谷歌还发布了哪些配套工具?

谷歌发布了用于应用迁移和修复程序错误的「TPU Agent Suite」,并更新了「LiteRT」以改善与PyTorch应用的集成。

来源