DeepSpeed

tracking optimizer step in cpu-adam when loading checkpoint

#564

Merged