AWS 发布 TorchNeuron 开发工具，让 PyTorch 应用一行配置跑上 Trainium

AWS 发布名为 TorchNeuron 的开发工具，让用 PyTorch 编写的 AI 应用只需改动一行配置即可运行在亚马逊自研芯片 Trainium 上，无需重写代码。亚马逊称约 85% 的 AI 应用用 PyTorch 编写，而 Nvidia 多年来让 CUDA 无缝支持 PyTorch，使其硬件几乎成为默认选项。AWS Annapurna Labs 技术软件副总裁 Colin Brace 表示，Trainium 的价值在于性价比以及训练和推理共用同一平台，并举例语音应用厂商 Deepgram 迁移后每美元处理的音频量提升 70%。亚马逊还表示将向上游 PyTorch 项目贡献代码，移除框架与特定芯片绑定的部分，使其对硬件更中立。原文也指出，AMD GPU 和 Apple 芯片此前已支持通过简单配置迁移 PyTorch 应用。