跳到正文
IT之家·· 1 天前同新闻AI 评分82

亚马逊开源 TorchNeuron 工具,PyTorch 代码无需重写即可在 Trainium 芯片运行

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

AI 导读

亚马逊推出开源工具 TorchNeuron,允许开发者将 PyTorch 代码在亚马逊自研 Trainium 芯片上本地运行,仅需修改一次配置参数,无需重写代码。

同一新闻,精选展示《TorchNeuron 让 HF 模型在 AWS Trainium 上运行更易》

来源:IT之家 · ithome.com