DeepSpeed

Refactor MoE and Groups API to simplify model creation and mangement

#1798

Merged