[DFM migration] Wan features verifications#2645

Merged

huvunvidia merged 23 commits intomigration/dfmfrom

migration/dfm_wan

Mar 13, 2026

Commits on Mar 4, 2026

worable code for checkpoint conversion, inferenceing, and training. No partial convergence test run
Huy Vu2
committed

Commits on Mar 10, 2026

with the M-LM dev PR (https://github.com/NVIDIA/Megatron-LM/pull/3759), rope fusion thd now works as expected
Huy Vu2
committed
just keep apply_rope_fusion for now'
Huy Vu2
committed
fix lint
Huy Vu2
committed

Commits on Mar 11, 2026

Commits on Mar 12, 2026

add dependencies
Huy Vu2
committed
uv lock
Huy Vu2
committed
fix unit tests
Huy Vu2
committed
fix unit tests
Huy Vu2
committed
fix lint
Huy Vu2
committed
adding back __subflavor__
Huy Vu2
committed
add functional tests for Wan, FLUX
Huy Vu2
committed
fix functional tests for wan, flux; cherry pick for - script: L2_Launch_recipes_gemma_vl
Huy Vu2
committed
restore all tests
Huy Vu2
committed

Commits on Mar 13, 2026

Merge remote-tracking branch 'origin/migration/dfm' into migration/dfm_wan
Huy Vu2
committed
fix conftest.py, update uv.lock
Huy Vu2
committed
update megatron.core.transformer.custom_layers.transformer_engine to megatron.core.extensions.transformer_engine
Huy Vu2
committed
lint
Huy Vu2
committed
fix jit expansion after Pytorch update
Huy Vu2
committed
removing, cleaning all redundant files
Huy Vu2
committed