DeepSpeed

Enable universal checkpoint for zero stage 1

#4516

Merged