DeepSpeed

engine.py - save_checkpoint: only rank-0 should create the save dir

#4536

Merged