首页 > 3D标签亚马逊推开源 TorchNeuron:改一处配置,PyTorch 就能跑在自研 Trainium 上 相关文章
亚马逊推开源 TorchNeuron:改一处配置,PyTorch 就能跑在自研 Trainium 上

亚马逊推开源 TorchNeuron:改一处配置,PyTorch 就能跑在自研 Trainium 上

亚马逊19日宣布推出开源工具TorchNeuron,将于20日在北美PyTorch大会正式开源。其核心卖点是PyTorch代码无需重写,仅改一处配置即可在自研Trainium芯片上原生运行。为验证“无缝”,工程师用同一AI训练脚本分别在传统芯片和Trainium上运行,唯一区别是指定芯片。