AWS 发布 TorchNeuron：Trainium 原生 PyTorch 后端

AWS 发布 TorchNeuron，一个面向 Trainium 的开源原生 PyTorch 后端，目标是让现有模型代码、库和工具无需改动即可在 Trainium2 和 Trainium3 上运行。AWS 称该后端以 Apache 2.0 许可证在 github.com/aws-neuron/torch-neuronx 开源，并从 2026 年 10 月 20 日起作为公开 Beta 版提供，开发者可通过 pip install torch-neuronx 从 AWS Neuron pip 仓库安装。 并非所有相关功能都已开放：AWS 表示 TorchAO 和 FP8 目前处于私有 Beta 阶段，将在下一版本中正式可用。AWS 还称正在 Neuron 上启用并验证相关集成，以把扩展的模型支持带到 Trainium 实例。

AI 根据报道生成 · 2 小时前更新

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