Fuse distillation + z-loss in the triton monolithic path (#507)#566

jlamypoirier merged 2 commits into

jlp_monolithic_head_lossServiceNow/Fast-LLM:jlp_monolithic_head_lossfrom

jlp_monolithic_distillation_zServiceNow/Fast-LLM:jlp_monolithic_distillation_zCopy head branch name to clipboard