Triton monolithic head-loss kernel (#507) - #558

Merged

jlamypoirier merged 6 commits into

jlp_monolithic_head_lossfrom

jlp_monolithic_triton

Jul 14, 2026

Commits on Jul 13, 2026

Triton monolithic kernel + triton policy metrics (#507)

jlamypoirier
and
claude
committed
2e9e206
Speed up GSPO metrics and single-loss masked rows in the monolithic triton path (#507)

jlamypoirier
and
claude
committed
b64da2b

Commits on Jul 14, 2026

Object-dispatch the triton monolithic loss driver (#507)

jlamypoirier
and
claude
committed
2bc48f7
Unit-test the monolithic triton kernel's GRPO and GSPO slots (#507)

jlamypoirier
and
claude
committed
3a9d813
Emit monolithic-triton policy metrics without a gradient (#507)

jlamypoirier
and
claude
committed
1a7f49d
Drop dead else-branch in monolithic triton kernel (#507)

jlamypoirier
and
claude
committed
513b557