perf(moe): reuse the DeepEP dispatch layout on activation-checkpoint recompute - #3684

Draft

akoumpa wants to merge 3 commits into

mainfrom

akoumparouli/moe-hybridep-ac-warning

Commits on Aug 26, 2026

fix(moe): warn that HybridEP dispatch is not reproducible under activation checkpointing
akoumpa
committed
aebce1d
perf(moe): reuse the DeepEP dispatch layout on activation-checkpoint recompute
akoumpa
committed
6644561
fix(moe): keep apply_ac working on models without nn.Module.modules()
akoumpa
committed
ee94057