亚马逊于当地时间 19 日宣布推出开源工具 TorchNeuron。它的卖点很直接:开发者写好的 PyTorch 代码,无需重写,只改一处配置,就能在亚马逊自研的 Trainium 芯片上原生跑起来。这套工具定于 20 日在加州圣何塞的北美 PyTorch 大会上以开源许可正式发布。
为了证明"无缝"不是嘴上说说,亚马逊工程师把同一份 AI 训练脚本分别跑在传统芯片和 Trainium 上,唯独区别是代码里指定所用芯片的设置参数。结果两个环境里,模型学到的内容完全一致、学习速度也相同。这意味着开发者手头那些用来测试、调试、扩展模型的工具,原封不动就能直接搬上 Trainium 用。
剑指 CUDA 垄断,要给开发者"换芯自由"
这套打法的对手,是英伟达经营了十多年的 CUDA 护城河。PyTorch 由 Meta 主导、被绝大多数 AI 开发者采用,而英伟达正是靠 CUDA 把 PyTorch 代码在自家芯片上跑得高效,把生态焊死在 GPU 上。亚马逊想撬动的,正是这层绑定。
亚马逊把 TorchNeuron 称作行业"长期缺失的关键要素"——真正可行的选择权。逻辑在于:当开发者能在不同芯片间自由切换而不必付出额外代价,芯片厂就只能用更具竞争力的产品去争宠;反过来,开发者则坐享成本下降与迭代提速。更进一步,亚马逊正主动参与 PyTorch 上游开发,与社区协作移除那些把框架和特定芯片绑定的代码,目标是让 PyTorch 对整个开发者群体更"硬件无关",而非只服务 Trainium 用户。
在算力争夺白热化的当下,亚马逊这一步既是为自家 Trainium 铺路,也是借开源松动英伟达的生态垄断。当"换芯"从伤筋动骨变成改行参数,GPU 一家独大的格局,便裂开了一道可被利用的缝。