Skip to content

Multimodal nemo gym compatible grpo pipeline - #3325

Closed
rohitrango wants to merge 18 commits into
mainfrom
rohit/gymv-mm-integration
Closed

Multimodal nemo gym compatible grpo pipeline#3325
rohitrango wants to merge 18 commits into
mainfrom
rohit/gymv-mm-integration

Conversation

@rohitrango

@rohitrango rohitrango commented Jul 23, 2026

Copy link
Copy Markdown
Contributor

What does this PR do ?

Adds multimodal, multiturn sync/async grpo with nemo gym environments.

Currently supports multi-image , multi-turn, tool calling on nemotron omni models (other models not verified).

Also contains some starter scripts.

Issues

List issues that this PR closes (syntax):

Usage

  • You can potentially add a usage example below
# Add a code snippet demonstrating how to use this

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.

Additional Information

mult prob errors and reward validation on toy environments:

  1. polygon naming (list name and color of polygon from 3 images - 2 images shown in turn 1 and 1 image shown in turn 2 via tool calls) - tests multi-turn tool calling capabilities
image
  1. doorkey env (multi-turn, async)
image
  1. tangram (single turn async)
image

aroshanghias-nvd and others added 18 commits July 15, 2026 16:59
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: Ali Roshan Ghias <aroshanghias@nvidia.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
Signed-off-by: rohitrango <rohit.rango@gmail.com>
@rohitrango
rohitrango requested review from a team as code owners July 23, 2026 17:30
@copy-pr-bot

copy-pr-bot Bot commented Jul 23, 2026

Copy link
Copy Markdown

Auto-sync is disabled for ready for review pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@aroshanghias-nvd
aroshanghias-nvd force-pushed the aroshanghias/nemotron-omni-main-migration branch from df0db31 to 875956c Compare July 29, 2026 14:18
@aroshanghias-nvd
aroshanghias-nvd requested review from a team as code owners July 29, 2026 14:18
Base automatically changed from aroshanghias/nemotron-omni-main-migration to main August 5, 2026 09:11
@rohitrango rohitrango closed this Aug 10, 2026
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