亚马逊TorchNeuron对抗Nvidia锁定的静默革命
亚马逊全新的TorchNeuron工具让PyTorch开发者只需一次配置更改,就能在其Trainium芯片上运行AI代码。但更深层的战略意图在于向上游进军,直接介入PyTorch本身,这可能会悄然瓦解Nvidia的生态系统优势。
真正的竞争早已不是芯片之争
亚马逊刚刚发布了 TorchNeuron,这是一个让 AI 开发者仅修改一行配置即可将 PyTorch 应用移植到其 Trainium 芯片的工具。表面上看,这是增量改进——又一个芯片厂商提供又一个兼容性层。但此次发布的架构透露出的才是真正的故事。
TorchNeuron 只是一个更深远举措的冰山一角。与此同时,亚马逊正在向 PyTorch 主项目本身贡献代码,努力消除将该框架绑定到 Nvidia 的硬件特定依赖。这不是在产品层面下注,而是在生态层面布局,并且瞄准的是 Nvidia 所构建的最持久的护城河——一种单纯靠性能数字无法瓦解的护城河。
人们对 AI 芯片战争的常规叙事聚焦于基准测试和每个 token 的定价。这种叙事是错的。真正的较量在于开发者习惯——那些潜藏在默认值、惯例和摩擦背后的隐形基础设施,决定了在使用之前,哪些硅片就已经被选中。
Nvidia 赢了这场较量。CUDA——其专有软件栈——如此无缝地融入了 PyTorch 和更广泛的 AI 开发工作流中,以至于选择 Nvidia 成了阻力最小的路径。据估计,如今 85% 的 AI 应用都在 PyTorch 上构建。多年来,这意味着 Nvidia 硬件。这种锁定效应从来不是因为规格更优,而是因为累积默认值的重量。
亚马逊正在攻击默认值
TorchNeuron 直接利用了那道长期将竞争对手拒之门外的单一摩擦点:迁移现有代码所需的成本。现在,将 PyTorch 应用移植到 Trainium 只需更改配置,而非重写代码。亚马逊表示,同一款芯片可同时处理训练和推理负载,且与 AWS 基础设施的集成意味着整个管线的总体拥有成本将得到改善。
据报道,AI 语音处理公司 Deepgram 已将其工作负载迁移至 Trainium,在达到生产级精度的同时,每美元的音频处理量提升了 70%。这是一个有意义的参考数据点,但它并非通用的证明。
亚马逊的论据不止于性价比。AWS Annapurna Labs——设计 Trainium 的部门——技术软件副总裁 Colin Brace 将价值主张建立在云关系本身之上。他表示,在 AWS 上构建意味着不必将赌注押在某一个专用计算提供商身上,然后再仓促地为其他一切构建替代方案。像 Coreweave、Nebius 和 Firmus 这样的新云计算厂商都与 Nvidia 硬件紧密绑定。在那些平台上起步的初创公司,可能在规模超出原始提供商的约束后,面临艰难的迁移抉择。
这一论点对创业者来说很有说服力。但不清楚这是否能真正压倒一个早已嵌入研究管线、学术课程和企业部署模式的 GPU 生态系统。
上游下注才是最危险的部分
更关键的一步或许是更安静的那一步。亚马逊正推动 PyTorch 在代码层面摒弃其 Nvidia 中心的假设。公司声明其正向 PyTorch 主项目贡献变更,以移除将框架与特定芯片绑定的代码,目标是在整个开发者社区中实现 PyTorch 的硬件中立。
AMD 和 Apple 都已为其芯片提供 PyTorch 兼容性,尽管易用程度各不相同。此处的区别在于,亚马逊并非仅仅为自家硬件构建一个兼容性层,而是在以一种惠及所有竞争芯片厂商的方式,将 PyTorch 从 Nvidia 的主导地位中解耦。
如果成功,这将是结构性转变。Nvidia 的护城河之所以得以强化,正是基于 PyTorch 与 CUDA 几乎不可分割的假设。打破这一假设,并不要求亚马逊推出更快的芯片,它只需要让 PyTorch 开发者的默认路径不再自动指向 Nvidia 硅片。
亚马逊面临的风险是,从 PyTorch 中移除硬件特定代码可能会给开发者引入新的复杂性。PyTorch 之所以能成功,其中一个原因就是它抽象化了让 AI 代码在实际硬件上运行的难度。让框架更具硬件中立性,这种抽象可能会变得不再那么流畅。Nvidia 已经受益于 CUDA 深度编织进 PyTorch 生态,提供了通用兼容性层可能无法复制的优化和支持。
亚马逊的上游贡献是否真的能解除这种依赖,仍然是一个开放性问题。代码变更已经公布,但它们在 PyTorch 社区中的反响以及在实际开发者工作流中的影响,尚未得到充分验证。
这对 AI 基础设施层级意味着什么
影响远不止亚马逊对阵 Nvidia。如果 PyTorch 真正实现了硬件中立,整个 AI 基础设施层级都将面临重新配置。难以获得市场接纳的芯片厂商——包括 AMD,以及 Groq 和 Cerebras 等新兴玩家——将从迁移门槛的降低中受益。将 Nvidia 硬件作为竞争优势的云厂商将面临利润率压力。初创公司将获得它们目前所缺乏的选项,而无需承担移植成本。
Nvidia 的回应将决定这股势头能否转化为市场份额的转移。该公司花了数年时间在 AI 开发的基础设施中嵌入 CUDA,但它并非停滞不前。新的软件举措和硬件路线图已定期发布。这些努力能否抵消框架层面的解耦,尚不清楚。
可以确定的是,战争已经超越了性能基准测试。亚马逊将战场转移到了 Nvidia 优势最深、也最难以察觉的层面:开发者每日书写代码的习惯,而这些代码最终将在别人的硬件上运行。
如果 TorchNeuron 和上游 PyTorch 的推动获得认可,这个故事就不会是关于更快芯片的,而是关于一个默认值的缓慢侵蚀。而一旦默认值发生转移,就极其难以逆转。
谁赢、谁输、接下来会发生什么
亚马逊赢了,前提是它能在不引发侵蚀 AWS 利润率的定价战的情况下,将一部分 PyTorch 开发者转向 Trainium。亚马逊将 Trainium 定位为总体成本优势,而非引人注目的性能优势。如果经济模型在实际负载中得到验证,这是一种可辩护的策略。
Nvidia 输了,前提是 PyTorch 的硬件中立成为标准做法。即使 AI 工作负载只有 modest 规模的迁移——从 Nvidia GPU 转向替代硅片——也会在高利润端压缩需求,而这一区域的利润率本就与维持领先地位所需的研发投入不成比例。
初创公司可能在短期内赢得最多。更多的硬件选择、更低的转换成本,以及一个不会将它们逼入单一供应商计算关系的 AWS 环境,都降低了早期阶段的风险。
未来几个季度的关键在于:PyTorch 的上游变更是否会被社区采纳,以及亚马逊能否证明 Trainium 不仅在孤立的案例研究中,而是在更广泛场景中交付一致且可泛化的成本优势。工具已上线,代码贡献已公布,竞争现在以开发者采纳率来衡量,而非基准测试结果。