feat: optimize get logprobs when cp enabled.#528

terrykong merged 9 commits into

mainNVIDIA-NeMo/RL:mainfrom

joyang/cp_optNVIDIA-NeMo/RL:joyang/cp_optCopy head branch name to clipboard