DeepSpeed

0bcd1551 - Merge branch 'master' into tohtana/model_declaration_in_init_context