Skip to content

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

Open
jlamypoirier wants to merge 27 commits into
mainfrom
jlp_monolithic_head_loss
Open

Monolithic fused head-loss kernel (torch.compile) (#507)#549
jlamypoirier wants to merge 27 commits into
mainfrom
jlp_monolithic_head_loss

Fuse GSPO into the monolithic compiled loss (#507) (#552)

96a6991
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar