transformers

Fix NaN weights on non-rank-0 FSDP processes

#45050

Merged