JIT for MoE router and preprocess - #1919

Merged

Victarry merged 11 commits into

NVIDIA:mainfrom

yaox12:xiny/jit_for_moe_main

Nov 19, 2025

Commits on Oct 24, 2025

use torch.compile
yaox12
committed
48edcb2
patch RandomSTE
yaox12
committed
d94eb19
fix
yaox12
committed
08dae18
add is_graph_capturing
yaox12
committed
bb20758
Merge branch 'main' into xiny/jit_for_moe_main
yaox12
committed
b3283b9

Commits on Oct 27, 2025

Merge branch 'main' into xiny/jit_for_moe_main
yaox12
authored
4f86bc5
update copyright
yaox12
committed
b802a61
update copyright
yaox12
committed
04b3623

Commits on Nov 18, 2025

Merge branch 'main' into xiny/jit_for_moe_main
yaox12
authored
88375e4
replace print with log
yaox12
committed
71dcec0
Merge branch 'main' into xiny/jit_for_moe_main
yanring
authored
6f50aab