IT之家·· 3 小时前精选AI 评分73
亚马逊推出开源工具 TorchNeuron，让 PyTorch 代码在 Trainium 芯片上本地运行
挑战英伟达 CUDA：亚马逊推出开源工具 TorchNeuron，自研 Trainium 芯片轻松运行 PyTorch 代码
AI 导读
亚马逊宣布推出开源工具 TorchNeuron，开发者只需修改一次配置即可让 PyTorch 代码在自研 Trainium 芯片上本地运行而无需重写。该工具将于北美 PyTorch 大会上正式开源，亚马逊同时正与 PyTorch 社区合作推动框架的硬件无关性。
推荐理由
原文介绍了亚马逊开源工具 TorchNeuron 的核心功能和测试表现，关注自研芯片生态的开发者可据此了解其代码迁移方式。
来源：IT之家 · ithome.com