Trainium3 实测:Qwen3-32B 达 30.47% MFU,编译 GPT-OSS 27 分钟才跑出第一步

demian_ai · x · 2026-11-03

转发的 SemiAnalysis 分析讨论 Amazon 自研 AI 芯片 Trainium 挑战 NVIDIA:难点不在芯片本身而在软件生态——几乎所有 AI 研究者用 PyTorch(生长于 NVIDIA GPU),新芯片必须把代码编译成自家硬件指令。

SemiAnalysis 用 AWS 新的 PyTorch 支持对 Trainium3 做了真实训练测试:

原因是历史包袱:Amazon 的编译器最初为推理构建,训练支持是后加的。Trainium 此前长于推理(Anthropic 已在 Trainium2/3 上跑生产推理与 RL rollout),SemiAnalysis 评价 AWS Neuron 团队转向一等公民的原生 PyTorch 训练体验(eager 模式 + torch.compile)方向正确,但成熟度仍远不及 CUDA。

原文链接 →

「Infra」频道最新

更多「Infra」频道 AI 资讯 →