fix(grpo-loss): compute group variance two-pass to avoid fp32 cancellation - #475

fix(grpo-loss): compute group variance two-pass to avoid fp32 cancellation#475

effintell wants to merge 4 commits into

RL-Align:mainRL-Align/RL-Kernel:mainfrom

effintell:fix/grpo-variance-cancellationeffintell/RL-Kernel:fix/grpo-variance-cancellationCopy head branch name to clipboard