Elastic Diffusion for Wan model#34

Open

huvunvidia wants to merge 6 commits into

NVlabs:mainfrom

huvunvidia:huvu/elastic_diffusion

Commits on Jul 13, 2026

Add elastification method design plan

Huy Vu
and
claude
committed

Commits on Jul 14, 2026

Trim PLAN.md to team-facing sections

Huy Vu
and
claude
committed

Commits on Jul 16, 2026

Add FlextronConfig + FlextronRouter for elastification

Huy Vu
and
claude
committed
Add FlextronRouter pytest suite

Huy Vu
and
claude
committed
Add budget_utils: analytical param count for Wan sub-networks

Huy Vu
and
claude
committed
Rewrite budget_utils.py to mirror Megatron structure; drop bad tests

Huy Vu
and
claude
committed