开源 TrnRT 引擎在单台 Trainium2 跑 GLM-5.3-Flash,每分钟 357 万 token

ycombinator · x · 2026-11-05

RunAnywhereAI 团队抱怨 AWS Trainium 对最新开源模型「零支持」,于是自行编写 megakernel 并开源了 Trainium 推理引擎 TrnRT。

核心成绩:在单台 Trainium2 服务器(TRN2.48XLARGE)上运行智谱 320B 的 GLM-5.3-Flash,达到 357 万 token/分钟,decode step 从 32.2ms 优化到 24.2ms(内存下限 6.1ms),数据来自 SemiAnalysis AgentX 的真实 coding agent 流量测试。

背景是 Anthropic 押注最高 5GW 的 AWS Trainium 算力,以及 AWS 的 Andy Jassy 宣布 TorchNeuron 让 PyTorch 代码一行配置即可原生跑在 Trainium 上——开源社区正在补齐这条芯片生态的推理短板。

所属事件:RunAnywhere 开源 Trainium 推理引擎 TrnRT,单机跑 320B GLM 模型(2 条相关)→

原文链接 →

「编程与Agent」频道最新

更多「编程与Agent」频道 AI 资讯 →