IT之家·· 1 天前同新闻AI 评分82
亚马逊开源 TorchNeuron 工具,PyTorch 代码无需重写即可在 Trainium 芯片运行
挑战英伟达 CUDA:亚马逊推出开源工具 TorchNeuron,自研 Trainium 芯片轻松运行 PyTorch 代码
AI 导读
亚马逊推出开源工具 TorchNeuron,允许开发者将 PyTorch 代码在亚马逊自研 Trainium 芯片上本地运行,仅需修改一次配置参数,无需重写代码。
同一新闻,精选展示《TorchNeuron 让 HF 模型在 AWS Trainium 上运行更易》
来源:IT之家 · ithome.com