NVIDIA-NeMo / NVIDIA-NeMo/RL

Add Support for Mcore-path in Reward Model Environment

Open
#1,154 0 comments 1 reaction 0 assignees View on GitHub
environments
Dominant language
Python
Stars
2k
Forks
561
Avg merge
4d 5h
Merged PRs (30d)
145

Description

## Description

The current reward model (RM) environment does not support advanced features such as mcore-path. By default, the RM environment uses the dtensor path, with dynamic batching and sequence packing disabled.

We should add support for these features to improve the efficiency and flexibility of the reward model. Support for the mcore-path will be considered for implementation after it is enabled for RM training.

If you also need support for the features in the RM environment, please click "👍" on this issue.

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.