Skip to content

Add fully_shard_optimizer for mixed-precision FSDP - #5411

Merged
wujingyue merged 12 commits into
NVIDIA:mainfrom
wujingyue:fsdp/optimizer
Jul 21, 2026
Merged

Add fully_shard_optimizer for mixed-precision FSDP#5411
wujingyue merged 12 commits into
NVIDIA:mainfrom
wujingyue:fsdp/optimizer

Relax MFSDP optimizer dtype assertion

542c84a
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Jul 20, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO