Skip to content

Make Megatron RL only materialize last token logit - #4551

Merged
tdene merged 1 commit into
NVIDIA:mainfrom
tdene:tde/fix_rl_materialize
Jun 16, 2026
Merged

Make Megatron RL only materialize last token logit#4551
tdene merged 1 commit into
NVIDIA:mainfrom
tdene:tde/fix_rl_materialize

Commits

Commits on Jun 16, 2026