Skip to content

feat: add MiniMax M2.7 support#2685

Open
jQizhang wants to merge 9 commits into
mainfrom
minimax-m2-rebase
Open

feat: add MiniMax M2.7 support#2685
jQizhang wants to merge 9 commits into
mainfrom
minimax-m2-rebase

Conversation

@jQizhang
Copy link
Copy Markdown
Contributor

@jQizhang jQizhang commented Jun 4, 2026

What does this PR do ?

Adds MiniMax M2.7 support for the Automodel DAPO/GRPO workflow.
Related issue: #2251

Summary of code changes:

  • nemo_rl/models/generation/vllm/quantization/fp8.py - update the vLLM FP8 Ray executor import path used by the local FP8 patching hook.
  • nemo_rl/models/policy/workers/dtensor_policy_worker_v2.py - call the optional Automodel MoE gate-bias update hook after each optimizer step.
  • examples/configs/recipes/llm/grpo-minimax-m27-dapo-8n8g-automodel.yaml - add an 8-node, 8-GPU MiniMax M2.7 DAPO recipe using Automodel, vLLM FP8 rollouts, and the DAPOMath datasets.

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests?
  • Did you run the unit tests and functional tests locally? Visit our Testing Guide for how to run tests
  • Did you add or update any necessary documentation? Visit our Document Development Guide for how to write, build and test the docs.

@jQizhang jQizhang requested review from a team as code owners June 4, 2026 14:18
@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Jun 4, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@jQizhang jQizhang requested a review from sharonyu-115 June 4, 2026 14:18
@sharonyu-115 sharonyu-115 added the CI:L1 Run doctests, unit tests, and functional tests label Jun 4, 2026
@sharonyu-115
Copy link
Copy Markdown
Contributor

/ok to test cb0ff8c

@jQizhang
Copy link
Copy Markdown
Contributor Author

jQizhang commented Jun 7, 2026

/ok to test 05caacb

@copy-pr-bot
Copy link
Copy Markdown

copy-pr-bot Bot commented Jun 7, 2026

/ok to test 05caacb

@jQizhang, there was an error processing your request: E2

See the following link for more information: https://docs.gha-runners.nvidia.com/cpr/e/2/

@jQizhang
Copy link
Copy Markdown
Contributor Author

jQizhang commented Jun 7, 2026

/ok to test 8cf2fa6

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

Labels

CI:L1 Run doctests, unit tests, and functional tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants