DeepSpeed

55ed1057 - fix bug related to stitching reduced grads across communication partitions (#318)