feat: R3 gym notq router replay - #2915
Conversation
|
/ok to test 763fb26 |
763fb26 to
c289b97
Compare
|
/ok to test c289b97 |
|
/ok to test df70c93 |
df70c93 to
f594c2d
Compare
|
/ok to test f594c2d |
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
2ff1ffc to
c25a0bf
Compare
|
/ok to test c25a0bf |
yuki-97
left a comment
There was a problem hiding this comment.
thanks @zyzhou5 , changes LGTM, some comments on test things.
can we add several nightly tests to guard r3 working correctly? we currently seem only cover grpo TQ path in examples/configs/recipes/llm/grpo-qwen3-30ba3b-8n8g-megatron-cp2-r3.yaml, could you add the below?
also maybe good to rename examples/configs/recipes/llm/grpo-qwen3-30ba3b-8n8g-megatron-cp2-r3.yaml to keep the same name style like the other TQ tests.
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
|
/ok to test 25c2f00 |
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
|
/ok to test 6b8df98 |
Signed-off-by: Zeyu Zhou <zezhou@nvidia.com>
|
/ok to test 43d34b5 |
Signed-off-by: Yuki Huang <yukih@nvidia.com>
|
/ok to test 58b6af1 |
|
What does this PR do ?
Adds no-TQ async Gym path support for R3 router replay.
When
policy.router_replay.enabled=true, async Gym rollouts now request routed expert indices from Gym, validate that they are present, and convert them into Nemo-RL message logs. The routed expert indices are sliced across prompt and generation tokens for each Gym model call so the existing R3 training path can consume them.This also updates the async OpenAI-compatible vLLM endpoint to return routed expert indices in chat response messages when requested.