DeepSpeed

773e454c - Merge branch 'reza/deepspeed_adam_merge_v3' into staging-zero-dual-v3