鲸云网10月20日消息，据媒体报道，亚马逊当地时间 10 月 19 日宣布推出开源工具 TorchNeuron，开发者编写的 PyTorch 代码可在其自研的 Trainium 芯片上原生运行，只需改动一处配置，无需重写。
TorchNeuron 将于 10 月 20 日在加州圣何塞举行的北美 PyTorch 大会上以开源许可形式正式发布。
PyTorch 是由 Meta 主导的开源深度学习框架，被大多数人工智能开发者所采用。英伟达则投入了十余年研发 CUDA 工具，让 PyTorch 代码得以在自家芯片上高效运行。
亚马逊工程师在常规芯片与 Trainium 上运行了同一套 AI 训练脚本，两者唯一的区别仅在于代码中指定所用芯片的配置参数。结果显示，模型在两种环境下学习到的内容完全一致，学习速度也相同，意味着开发者原本用于测试、调试和扩展 AI 模型的工具无需任何修改即可直接在 Trainium 上使用。
亚马逊方面表示，TorchNeuron 为人工智能行业补上了长期缺失的关键要素——真正可行的选择权。当开发者能够在不同芯片之间自由切换而无需付出额外代价时，芯片厂商就必须以更具竞争力的产品来争取开发者，开发者也能从中获得成本下降与迭代提速的收益。
此外，亚马逊正参与 PyTorch 上游开发，与社区合作移除将框架与特定芯片相绑定的代码，目标是让 PyTorch 对全体开发者更具通用性。
编辑：小昭
免责声明：本站转载的文章，版权归原作者所有；旨在传递信息，不代表本站的观点和立场。
