跳到正文
AWS Machine Learning Blog· Mona Mona·· 5 小时前AI 评分40

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

Mixture-of-Experts kernel engineering on Amazon SageMaker AI

阅读原文

本站未展示全文,请前往来源网站阅读。

AI 导读

针对 MoE 模型在解码阶段因 M 维过小导致的 GPU 利用率瓶颈,团队通过 Swap-AB 转置将专家维度 N=8192 提升为首操作数,在 H200 上实现端到端输出吞吐提升 2–4%,该优化已合入 vLLM 0.24.0 及更高版本。

来源:AWS Machine Learning Blog · aws.amazon.com