Skip to content
AWS Machine Learning Blog· Ashvin Nihalani·· 23 days agoAI score31

Scaling MoE reinforcement learning on Amazon EKS with EFA and DeepEP with 40% more throughput

Scaling MoE reinforcement learning on Amazon EKS with EFA and DeepEP with 40% more throughput

AI introduction

AWS proposed an architecture based on Amazon EKS, EFA and DeepEP to accelerate large-scale reinforcement learning post-training of MoE models, delivering 40% higher throughput. The solution targets workloads like RLHF, PPO and GRPO, addressing resource balancing between rollout generation and policy training, high-throughput communication across hundreds of accelerators, and the dynamic all-to-all token routing overhead introduced by Expert Parallelism.

A summary is available. Open the original source for the full article. Read source
Source images · Select to enlarge

Source: AWS Machine Learning Blog · aws.amazon.com