DeepSpeed

4e748d0b - Merge branch 'master' into tohtana/model_declaration_in_init_context