挑战英伟达CUDA,亚马逊推出开源工具TorchNeuron
当地时间10月19日,亚马逊宣布推出开源工具TorchNeuron,并于20日在北美PyTorch大会上正式发布。该工具使开发人员编写的PyTorch代码仅需更改一次配置,即可在亚马逊自研Trainium芯片上本地运行,无需重写代码。测试显示,AI模型在Trainium与传统芯片上的训练效果及速度一致,原有调试工具可直接复用。亚马逊表示,此举旨在为开发者提供芯片选择权以降低成本并提升迭代速度。此外,亚马逊正与PyTorch社区合作移除特定芯片绑定代码,推动框架实现“硬件无关性”。

