DeepSpeed

d26c258b - added shaden's set_train_batch_size patches, plus formatting