DeepSpeed

Samyamr/full precision for ZeRO Stage2 and Stage3

#1004

Merged