transformers

[`distributed`] Divide the expert gradients once when `efsdp_size == 1`

#50292

Merged