Skip to content

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

Open
jlamypoirier wants to merge 31 commits into
mainfrom
jlp_monolithic_head_loss
Open

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

Policy metrics in the compiled monolithic loss (#507) (#565)

18c4662
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar