Don't compute the logprobs and their gradients for tokens no loss com… - #3326

Don't compute the logprobs and their gradients for tokens no loss com…#3326

MarioSieg wants to merge 7 commits into

mainPrimeIntellect-ai/prime-rl:mainfrom

mario/optimize-masked-tokenlossPrimeIntellect-ai/prime-rl:mario/optimize-masked-tokenlossCopy head branch name to clipboard