| 亚马逊于当地时间19日宣布推出开源工具TorchNeuron,该工具允许开发者无需重写PyTorch代码,仅通过修改一处配置即可在亚马逊自研的Trainium芯片上运行。该工具定于20日在加州圣何塞的北美PyTorch大会上以开源许可正式发布。亚马逊工程师测试了同一份AI训练脚本在传统芯片和Trainium上的运行,结果显示模型学习内容和速度一致。亚马逊表示,TorchNeuron旨在提供芯片间的可移植性,并正参与PyTorch上游开发,以移除框架与特定芯片绑定的代码。 |
AI INSIGHT
亚马逊推开源 TorchNeuron:改一处配置,PyTorch 就能跑在自研 Trainium 上
AI 摘要
亚马逊于当地时间19日宣布推出开源工具TorchNeuron,该工具允许开发者无需重写PyTorch代码,仅通过修改一处配置即可在亚马逊自研的Trainium芯片上运行。该工具定于20日在加州圣何塞的北美PyTorch大会上以开源许可正式发布。亚马逊工程师测试了同一份AI训练脚本在传统芯片和Trainium上的运行,结果显示模型学习内容和速度一致。亚马逊表示,TorchNeuron