跳到正文
IT之家·· 20 天前AI 评分72

亚马逊推出开源工具 TorchNeuron,让 PyTorch 代码在 Trainium 芯片上运行

挑战英伟达 CUDA:亚马逊推出开源工具 TorchNeuron,自研 Trainium 芯片轻松运行 PyTorch 代码

AI 导读

亚马逊宣布推出开源工具 TorchNeuron,开发者只需更改一次配置,无需重写代码,即可让 PyTorch 代码在自研 Trainium 芯片上本地运行,该工具将于当地时间 10 月 20 日在北美 PyTorch 大会上以开源许可形式正式发布。

来源:IT之家 · ithome.com