DeepSpeed

e8318634 - Spread layers more uniformly when using partition_uniform (#4053)

Spread layers more uniformly when using partition_uniform (#4053) * update partition_uniform util function * formatting --------- Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>