Add MambaLayer FSDP support and fix TP annotation for SSM parameters#4329

Closed

Phlip79 wants to merge 2 commits intoNVIDIA:mainfrom

Phlip79:worktree-agent-a7696c2d

Commits on Apr 15, 2026

Add MambaLayer FSDP support and fix TP annotation for SSM parameters

Phlip79
and
claude
committed

Commits on Apr 24, 2026

Merge remote-tracking branch 'upstream/main' into worktree-agent-a7696c2d
Phlip79
committed