DeepSpeed

bf53561c - Adding static_scale test back for Z3, and also changing hidden size to be not divisile by world_size