跳到正文
AWS Machine Learning Blog· Mona Mona·原文发布· 2 天前AI 评分30

Amazon SageMaker AI 上的混合专家(MoE)内核工程实践

Mixture-of-Experts kernel engineering on Amazon SageMaker AI

AI 导读

针对 MoE 模型在解码阶段 M 维过小导致 Hopper GPU 利用率低的问题,AWS 通过 Swap-AB 转置让专家维度 N=8192 进入首个操作数位置,在 H200 上实现端到端输出吞吐提升 2–4%,该优化已合入 vLLM 0.24.0 及之后版本。

本站提供 AI 导读,完整报道请前往来源网站阅读。

阅读原文 AWS Machine Learning Blog · aws.amazon.com