Trainium 是亚马逊云科技（AWS）自研的 AI 训练芯片系列，主要用于大规模机器学习模型训练，是英伟达 GPU 之外的重要替代算力选择。2026 年 10 月，随着 TorchNeuron 开发工具的推出，Trainium 正式获得 PyTorch 原生支持，开发者无需修改代码即可将现有模型迁移到这颗芯片上运行，这被业内视为亚马逊对英伟达发起的一次重要挑战。
Trainium
general · First seen 2026-05-22 · Last seen 2026-10-23 · 81 mentions
Overview
Coverage
10 on file- AWS Trainium 芯片推出 PyTorch 后端 TorchNeuronPyTorch
- AWS 开源 TorchNeuron，PyTorch 代码可零改动跑在 Trainium 上Amazon Science
- Google 开源 TorchTPU，让 PyTorch 原生跑在 TPU 芯片上Blognone
- AWS Trainium 模型支持查询：torch-neuronx 兼容性一览Hugging Face: Blog
- PyTorchCon 2026：Google、AWS 分别推出 TorchTPU 和 TorchNeuronPyTorch
- SGLang 适配 AWS Trainium，基于 TorchNeuron 推出 Beta 版LMSYS Org (SGLang)
- PyTorchCon Demo Theater 13 场演示：从 Trainium 强化学习到 MoE 推理PyTorch
- 在 AWS Trainium 上用 torchtitan 训练 Qwen3 的三种方式AWS Machine Learning Blog
- TorchNeuron 性能剖析教程：在 AWS Trainium 上做系统级 profilingHugging Face: Blog
- 谷歌开源 TorchTPU，让 PyTorch 应用免重写迁移到 TPUIT之家