update jax/flax version#274

Merged

susanbao merged 5 commits intomainfrom

Oct 22, 2025

Commits on Oct 21, 2025

update jax/flax version
susanbao
committed

Commits on Oct 22, 2025

update script for WAN 2.1
susanbao
committed
HIDDEN_STATE_WITH_OFFLOAD has not been supported
susanbao
committed
fix
susanbao
committed
update
susanbao
committed