亚马逊推出 TorchNeuron,让 PyTorch 代码无需重写即可在 Trainium 芯片上运行
亚马逊推出开源工具 TorchNeuron,旨在让开发者无需重写代码,就能在亚马逊的 Trainium AI 芯片上运行用 PyTorch 编写的代码。该工具希望简化从 Nvidia 硬件转向亚马逊自有芯片的过程。
本文来源:Techmeme,仅供学习参考,版权归原作者所有。