DeepSpeed

9f52a36f - tracking optimizer step in cpu-adam when loading checkpoint (#564)

tracking optimizer step in cpu-adam when loading checkpoint (#564) * tracking optimizer step in cpu-adam when loading checkpoint * add warning/error message for updating optimizer step count * resolve build issue * supporting state update from the python side * track step from python in all cases * remove comma