Skip to content

[JAX] Fix FSDP when FSDP+EP is active#2649

Merged
jberchtold-nvidia merged 4 commits intoNVIDIA:mainfrom
jberchtold-nvidia:jberchtold/fix-fsdp-when-ep-is-active
Feb 24, 2026
Merged

[JAX] Fix FSDP when FSDP+EP is active#2649
jberchtold-nvidia merged 4 commits intoNVIDIA:mainfrom
jberchtold-nvidia:jberchtold/fix-fsdp-when-ep-is-active

Commits

Commits on Feb 4, 2026