Raw aux loss sum was 11M (unnormalized across B×K positions). Now divides by n_valid per direction (prof/size), matching the original step_batched normalization. aux: 11M → 116. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
Raw aux loss sum was 11M (unnormalized across B×K positions). Now divides by n_valid per direction (prof/size), matching the original step_batched normalization. aux: 11M → 116. Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>