谷歌本周宣布，其备受期待的PyTorch版本 TorchTPU 已作为开源项目发布。前一天，亚马逊也发布了类似的 TorchNeuron。两者都试图让AI应用更容易运行在非英伟达硬件上，从而松动英伟达在AI芯片市场的主导地位。

PyTorch是AI开发者广泛使用的开源框架，据称约 85% 的AI应用基于它。它由Meta在2016年创建，当时英伟达GPU已是AI研究的标准硬件，因此PyTorch的许多特性默认适配英伟达。2022年，Meta将PyTorch移交给Linux基金会。这种历史绑定让英伟达在AI开发者生态中占据天然优势。

TorchTPU允许现有PyTorch应用通过一次简单的配置变更迁移到谷歌TPU和Google Cloud，无需重写代码。TorchNeuron则支持将应用移植到亚马逊自研的Trainium芯片和AWS。亚马逊还向PyTorch主项目提交代码，试图将部分与英伟达绑定的功能解耦，使其默认支持更多硬件。

谷歌似乎更进一步。它称已与Meta、Hugging Face、Inferact（vLLM开发者）、RadixArk（SGLang开发者）等合作，确保TorchTPU与这些流行工具链的兼容性。同时还发布了 TPU agent suite 来帮助开发者迁移与排错，并更新了LiteRT——用于在手机、笔记本和嵌入式设备上运行AI的工具——使其与PyTorch AI应用更好地配合。

这并不意味着AI开发者会立刻大批抛弃英伟达硬件，但亚马逊和谷歌显然在努力终结英伟达的这项优势。