RecodeX 重构消息,亚马逊推出TorchNeuron软件,开发者只需做基础配置修改,即可将基于开源框架PyTorch编写的代码运行在其自研Trainium芯片上。该工具目前以公开测试版形式向所有开发者开放,并同步开源发布。此举意在削弱英伟达在AI软件生态上的优势。
亚马逊开源TorchNeuron,让PyTorch代码直跑Trainium芯片
后续报道
亚马逊推出TorchNeuron测试版简化Trainium芯片PyTorch部署
RecodeX 重构消息,亚马逊发布TorchNeuron测试版工具,旨在降低在自研Trainium芯片上运行PyTorch的难度。该测试版支持Trainium2和Trainium3芯片,但经编译的模型需要固定输入形状。