DeepSpeed

[zero-3] shutdown zero.Init from within ds.init

#2150

Merged