据外媒报道，亚马逊于当地时间10月19日发布开源工具TorchNeuron，开发者编写的PyTorch代码无需重写，只需更改一次配置，就能在亚马逊自研的Trainium芯片上本地运行。
该工具将于当地时间10月20日在加州圣何塞举行的北美PyTorch大会上以开源许可形式正式发布。PyTorch由Meta主导，被大多数人工智能开发者采用；英伟达则投入十余年打造CUDA工具，让PyTorch代码在其芯片上高效运行。
亚马逊工程师曾在传统芯片和Trainium上运行相同的AI训练脚本，唯一的区别是代码中指定所用芯片的设置参数。结果显示，模型在两种环境下学到的内容完全一致，学习速度也相同，这意味着开发者用于测试、调试和扩展模型的工具无需任何修改，即可直接在Trainium上使用。
亚马逊称，TorchNeuron为行业带来了一项长期缺失的关键要素，即真正可行的选择权。当开发者能在不同芯片之间自由切换而不必付出额外代价时，芯片厂商就必须靠更有竞争力的产品赢得客户，开发者则能从中受益，实现成本降低和迭代加速。
与此同时，亚马逊正积极参与PyTorch上游开发，与社区合作移除把框架与特定芯片相绑定的代码，目标是让PyTorch对整个开发者社区更具硬件无关性，而不只是服务Trainium用户。
原创文章，作者：李小白，如若转载，请注明出处：https://www.kejixun.co/article/765206.html
