跳到正文
AWS Machine Learning Blog· Mona Mona·· 2 小时前精选AI 评分64

AWS 在 Amazon SageMaker AI 上的 MoE 内核工程优化

Mixture-of-Experts kernel engineering on Amazon SageMaker AI

AI 导读

AWS 在 Amazon SageMaker AI 的 LMI 容器中应用多项 fused MoE 内核优化,使吞吐最高提升 10%,TTFT 最高加快 9%。

推荐理由

文章把 Hopper 的 WGMMA 布局、FP8 后端路由和 Triton JIT 编译分别对应到可操作的优化点,适合参考 fused MoE 推理的性能调优路径。

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