AWS 开源 Trainium 原生 PyTorch 后端 TorchNeuron

AWS 发布 TorchNeuron，一个面向 Trainium 芯片的开源原生 PyTorch 后端，使现有的模型代码、库和工具可直接在 Trainium 上运行。该后端以 Apache 2.0 协议开源于 github.com/aws-neuron/torch-neuronx，自 2026 年 10 月 20 日起进入公开 Beta。 研究人员和开发者可通过 AWS Neuron pip 源执行 pip install torch-neuronx 安装。TorchAO 与 FP8 目前处于私有 Beta 阶段，AWS 称将在下一个版本中转为正式可用。

AI 根据报道生成 · 45 分钟前更新

还没有足够的连续观测数据，暂不绘制趋势。