搜索
亚马逊推出开源工具 TorchNeuron，可让 PyTorch 代码在自研 Trainium 芯片运行
1 天前
亚马逊推出开源工具TorchNeuron，开发者仅需修改一次配置，无需重写代码，即可让原有PyTorch代码在其自研的Trainium芯片上本地运行。该工具将在北美PyTorch大会上正式开源。实测显示，相同的AI训练脚本在传统芯片与Trainium芯片上运行效果相同，原有开发工具可直接在Trainium上使用。TorchNeuron为行业提供了更多芯片选择，有助于推动芯片厂商提升产品竞争力，降低开发者成本，提高效率。此外，亚马逊正参与PyTorch上游开发，推动框架实现硬件无关性。
