跳到正文
IT之家·· 2 小时前

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

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

AI 导读

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

来源:IT之家 · ithome.com