返回全部动态

AWS 在 EKS 上用 EFA 和 DeepEP 扩展 MoE 强化学习,吞吐量提升 40%

原标题:Scaling MoE reinforcement learning on Amazon EKS with EFA and DeepEP with 40% more throughput

AWS Machine Learning Blog一手来源教程质量 60

AI 摘要

AWS 机器学习博客介绍了一种在 Amazon EKS 上结合 EFA 与 DeepEP 扩展 MoE 强化学习训练的架构,声称吞吐量提升 40%。文章指出大规模 RLHF/GRPO 训练面临 rollout 生成与策略训练资源竞争、计算/内存/带宽平衡、以及从节点内 NVLink 到节点间低带宽通信转变三大挑战,其中专家并行(EP)的动态 all-to-all 通信是主要瓶颈。该方案通过 EKS 编排、EFA 加速网络和 DeepEP 优化专家并行通信来应对这些问题。

以上摘要由 AI 生成,可能存在误差。事实请以原文为准。

正文节选

Scaling MoE reinforcement learning on Amazon EKS with EFA and DeepEP with 40% more throughput When you post-train a Mixture-of-Experts (MoE) model with Reinforcement Learning from Human Feedback (RLHF) or Group Relative Policy Optimization (GRPO) at scale, three simultaneous challenges emerge. The first requires coordinating heterogeneous compute for rollout generation and policy training. Second, sustaining high-throughput communication across hundreds of accelerators. And third, dynamically or


发布时间:2026-09-26 00:29
抓取时间:2026-09-26 01:10
来源机构:AWS
阅读原文aws.amazon.com