据TechCrunch AI报道，谷歌于周二宣布，其等待已久的PyTorch版本TorchTPU已作为开源项目发布。谷歌由此加入亚马逊的行列，试图打破英伟达在AI行业的主导地位——目前多数AI应用默认运行在英伟达的GPU上。

PyTorch是AI应用开发者和初创公司用于构建、训练和运行AI模型的开源工具包，有说法称其被85%的AI应用使用。它于2016年由Meta（当时为Facebook）开发，当时英伟达的GPU已是AI研究的标准硬件，因此PyTorch的许多功能是专门为英伟达芯片构建的。2022年，Meta将PyTorch移交给Linux基金会。

TorchTPU可让AI应用运行在谷歌称为TPU的AI芯片上，也便于将现有PyTorch应用迁移到TPU，进而迁移到谷歌云。开发者只需更改配置，无需重写应用。

亚马逊周一发布了类似消息，推出自己的PyTorch版本TorchNeuron。TorchNeuron可让AI应用在不重写的情况下迁移到亚马逊自研AI芯片Trainium和AWS。亚马逊还表示，它正向PyTorch主项目提交代码，将多项功能与英伟达芯片解耦，使PyTorch默认对任何硬件更加开放。

谷歌表示，它在几个方面做了额外安排。谷歌称，它直接与Meta、Hugging Face、Inferact（vLLM开发商）和RadixArk（SGLang开发商）等合作，确保TorchTPU的功能和代码库能像原版PyTorch一样适配这些流行的AI开发工具。谷歌还推出了专用工具TPU agent suite，帮助开发者将应用迁移到谷歌芯片和云平台，并修复迁移过程中的错误。它甚至更新了在手机、笔记本电脑和嵌入式设备上运行AI的LiteRT工具，使其更好地支持PyTorch AI应用。

这些动作并不意味着AI应用开发者会因为现在可以更容易地转向替代芯片和云平台，就自动大规模放弃英伟达硬件。但亚马逊和谷歌显然正在努力终结英伟达的优势。