[fix][train] Fix advantage response masking for step wise training #1506

Closed

SumanthRH wants to merge 2 commits intomainfrom

fix-adv-masking-step-wise

Commits on Apr 13, 2026

[fix][train] Fix advantage masking for step-wise trajectories (#1492)

SumanthRH
and
claude
committed
Add GAE return_raw_scores support and unify estimator output shapes to [N, seqlen]

SumanthRH
and
claude
committed