Added named_scope for WAN 2.1 Xprof profiling#285

Merged

coolkp merged 4 commits intomainfrom

elisatsai-wan2.1-xprof

Nov 21, 2025

Commits on Nov 19, 2025

Added named_scope for WAN 2.1 Xprof profiling
eltsai
committed

Commits on Nov 20, 2025

Added flag to enable named_scope
eltsai
committed
Make named_scope in flash_attention respect enable_jax_named_scopes flag
eltsai
committed

Commits on Nov 21, 2025

Added named scope for WanModel output
eltsai
committed