谷歌开源TorchTPU,允许AI应用无需重写代码即可在TPU上运行原生PyTorch
谷歌发布TorchTPU开源项目,使开发者能通过简单配置将现有PyTorch应用迁移至TPU和谷歌云,旨在打破英伟达在AI芯片领域的垄断。
重要性实质性证据E3 可检验写法标准
2026年10月20日,谷歌(Google)宣布其长期期待的TorchTPU项目现已作为开源软件发布。该项目是PyTorch的一个版本,允许人工智能应用程序直接在谷歌的张量处理单元(TPU)和谷歌云上运行,而无需重写代码。这一举措被视为谷歌与亚马逊联手,试图打破英伟达(Nvidia)在AI行业主导地位的关键一步。
PyTorch是目前最流行的开源AI开发工具包,据估计被85%的AI应用使用。它最初由Meta于2016年开发,当时英伟达的GPU已是AI研究的标准硬件,因此PyTorch的许多功能都是为配合GPU而构建的。2022年,Meta将PyTorch移交给Linux基金会管理。TorchTPU通过提供简单的配置更改,降低了开发者将现有应用从GPU迁移到TPU及谷歌云的门槛。
就在前一天(10月19日),亚马逊也发布了类似的工具TorchNeuron,支持将AI应用移植到其自研芯片Trainium和AWS上。此外,亚马逊还向主PyTorch项目提交代码,以解耦特定功能与英伟达芯片的绑定,使框架默认对更多硬件开放。谷歌方面表示,其与Meta、Hugging Face以及vLLM和SGLang的制造商等合作,确保TorchTPU的功能和库能与这些流行开发工具良好兼容。同时,谷歌推出了专用的“TPU代理套件”帮助开发者迁移应用并修复错误,并更新了LiteRT工具以更好地支持PyTorch应用。