Amazon SageMaker AI 上的 MoE kernel 工程优化实践

AWS 在 Amazon SageMaker AI 上为 fused MoE 模型实现 kernel 优化：通过 Swap-AB 转置让专家维度 N=8192 进入 WGMMA 首个操作数位置，在 H200 上带来 2–4% 端到端输出吞吐提升，该优化已合入 vLLM 0.24.0 及之后版本。