Fix nan leak from masked positions in `compute_approx_kl`#1635

Merged

erictang000 merged 2 commits into

NovaSky-AI:mainfrom

EdisonScientific:fix-compute-approx-kl-nan-leak

May 8, 2026

Commits on May 8, 2026

[bug][train] Fix nan leak from masked positions in compute_approx_kl

jamesbraza
and
claude
committed
[bug][train] Preserve soft-mask scaling in compute_approx_kl

jamesbraza
and
claude
committed