IT之家·· 1 天前
亚马逊推出开源工具 TorchNeuron,Trainium 芯片可直接运行 PyTorch 代码
挑战英伟达 CUDA:亚马逊推出开源工具 TorchNeuron,自研 Trainium 芯片轻松运行 PyTorch 代码
SI 导读
亚马逊推出开源工具 TorchNeuron,使 PyTorch 代码无需重写即可在自研 Trainium 芯片上本地运行。测试显示,在 Trainium 与传统芯片上运行相同 AI 训练脚本,模型学习结果与速度完全一致。此举旨在打破英伟达 CUDA 的垄断,为开发者提供跨芯片自由切换的选择权。
SI 评分49
来源:IT之家 · ithome.com