Skip to content

perf(loss): defer the fp32 upcast in the distillation top-k path - #3496

Open
tianyi-zhang-02 wants to merge 16 commits into
NVIDIA-NeMo:mainfrom
tianyi-zhang-02:perf/distill-topk-defer-fp32-upcast
Open

perf(loss): defer the fp32 upcast in the distillation top-k path#3496
tianyi-zhang-02 wants to merge 16 commits into
NVIDIA-NeMo:mainfrom
tianyi-zhang-02:perf/distill-topk-defer-fp32-upcast

Merge upstream/main into w3496

7534040
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 30, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO