Skip to content

feat(slurm): support single-node Dynamo RL jobs - #3071

Closed
furionw wants to merge 20 commits into
PrimeIntellect-ai:mainfrom
furionw:feat/dynamo-slurm-single-node
Closed

feat(slurm): support single-node Dynamo RL jobs#3071
furionw wants to merge 20 commits into
PrimeIntellect-ai:mainfrom
furionw:feat/dynamo-slurm-single-node

Conversation

@furionw

@furionw furionw commented Jul 17, 2026

Copy link
Copy Markdown

Based on #2964.

Why

#2964 adds a Dynamo inference backend, but a top-level [slurm] currently propagates into the nested inference config, where Dynamo is rejected. Single-node RL does not need a separate inference submission: after Slurm assigns the node, the batch job can enter Prime's existing local Dynamo launcher. This enables scheduled aggregate and P/D workflows while preserving the rejection for standalone inference and multi-node Dynamo deployments.

What Change

  • Keep single-node Dynamo local inside Prime-managed Slurm allocations.
  • Preserve Dynamo worker topology in generated inference configs.
  • Add aggregate and P/D examples, documentation, and regression coverage.

Test Plan

  • Run focused config and RL entrypoint unit tests.
  • Complete three-step aggregate and P/D workflows on one four-GPU Slurm node.

Validated with ai-dynamo/dynamo#11254.

biswapanda and others added 20 commits July 11, 2026 03:28
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Signed-off-by: Biswa Panda <biswa.panda@gmail.com>
Co-Authored-By: Claude <noreply@anthropic.com>
@furionw
furionw force-pushed the feat/dynamo-slurm-single-node branch from 91cb79b to 61484f5 Compare July 17, 2026 19:23
@furionw furionw closed this Jul 17, 2026

furionw commented Jul 17, 2026

Copy link
Copy Markdown
Author

Replaced by furionw/prime-rl#3.

This work moved to a fork-local stack so upstream #2964 remains authoritative while the single-node and multi-node layers each have an exact review base. This PR remains closed, and its feat/dynamo-slurm-single-node branch is retained as the base of the multi-node layer.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants