DeepSpeed

2b985e07 - Merge branch 'master' into loadams/fix-fp16-bf16-logging-issue