软件生态正成为AI芯片竞争的关键战场。亚马逊于当地时间10月19日推出开源工具TorchNeuron,旨在降低开发者从英伟达芯片转向其自研芯片的门槛,挑战英伟达依托CUDA构筑的软件壁垒。
PyTorch是Meta主导维护的开源AI编程框架,被大多数开发者视为AI开发领域的“通用语言”,大量代码目前运行在英伟达芯片上。据亚马逊旗下芯片设计部门Annapurna Labs介绍,借助TorchNeuron,开发者只需修改一项配置,同一份代码就能直接在亚马逊自研的Trainium芯片上运行,无需重写。
该工具瞄准的正是英伟达的核心优势。英伟达花费十余年打磨CUDA软件工具,让PyTorch代码在其芯片上高效运行,开发者换用其他芯片往往需要额外适配,迁移成本较高。Annapurna Labs软件副总裁Colin Brace表示,推出TorchNeuron旨在减少开发者试用亚马逊芯片所需的步骤,“从客户现有的工作方式出发”。
目前已有超过30家早期客户使用该软件,覆盖机器人、世界模型和语音AI等领域,转用亚马逊芯片可带来大幅成本节约。该工具计划于10月20日在加州圣何塞举行的PyTorch大会上以免费开源形式公开发布。
