Skip to content

Monolithic fused head-loss kernel (torch.compile) (#507)#549

Merged
jlamypoirier merged 37 commits into
mainfrom
jlp_monolithic_head_loss
Jul 20, 2026
Merged

Monolithic fused head-loss kernel (torch.compile) (#507)#549
jlamypoirier merged 37 commits into
mainfrom
jlp_monolithic_head_loss

Review fixes: widen data-metrics type hint, name regularized log-sum-…

82f95bc
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar