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

IT之家 10 月 20 日消息,当地时间 10 月 19 日,亚马逊宣布推出开源工具 TorchNeuron,借助该工具,开发人员编写的 PyTorch 代码可以在亚马逊的自研 Trainium 芯片上本地运行, 只需更改一次配置,无需重写 。 IT之家获悉,TorchNeuron 将于当地时间 10 月 20 日在加州圣何塞举行的北美 PyTorch 大会上以开源许可形式正式发布。 PyTorch 是由 Meta 主导的开源框架,被大多数人工智能开发者所采用。英伟达投入了十多年的时间研发 CUDA 工具,能够让 PyTorch 代码在英伟达的芯片上高效运行。 据介绍,亚马逊工程师分别在传统…
以上为资讯摘要,完整内容请访问原文网站。
阅读原文 →