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