跳到正文
IT之家·· 5 天前精选AI 评分76

亚马逊推出开源工具 TorchNeuron,支持自研 Trainium 芯片运行 PyTorch 代码

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

AI 导读

亚马逊宣布推出开源工具 TorchNeuron,允许开发者编写的 PyTorch 代码只需更改一次配置即可在自研 Trainium 芯片上本地运行,无需重写代码。该工具将于 10 月 20 日在北美 PyTorch 大会上以开源许可形式发布,旨在为开发者提供除英伟达 CUDA 外的可行选择。

推荐理由

开源工具让 PyTorch 代码无需重写即可在自研芯片上运行,挑战英伟达 CUDA 垄断地位。

来源:IT之家 · ithome.com