谷歌开源TorchTPU，加入亚马逊挑战英伟达AI芯片主导地位

OmniTools 10月21日消息，据TechCrunch报道，谷歌宣布其长期开发的PyTorch版本TorchTPU已作为开源项目发布。该工具可让AI应用在谷歌张量处理单元（TPU）上运行，开发者只需修改配置即可将现有PyTorch应用迁移至TPU及谷歌云，无需重写代码。

此前一天，亚马逊发布了类似工具TorchNeuron，可将AI应用迁移至其自研芯片Trainium和AWS，同样无需重写。亚马逊还表示将向PyTorch主项目提交代码，使相关功能与英伟达芯片解耦。谷歌称，TorchTPU与Meta、Hugging Face、Inferact（vLLM）、RadixArk（SGLang）等合作，确保代码库兼容，并推出TPU agent suite帮助迁移和修复错误，同时更新LiteRT以更好支持PyTorch应用。

PyTorch由Meta于2016年开发，2022年交由Linux基金会管理。有说法称，约85%的AI应用使用PyTorch。多数AI应用默认运行在英伟达GPU上，谷歌和亚马逊的最新动作意在削弱英伟达在AI硬件领域的主导地位。