软件生态正成为 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 等领域。他表示,转用亚马逊芯片可为客户带来大幅成本节约。