DeepSpeed

c0af6d90 - Refactor MoE and Groups API to simplify model creation and mangement (#1798)

Refactor MoE and Groups API to simplify model creation and mangement (#1798) Co-authored-by: yaozhewei <zheweiy@berkeley.edu> Co-authored-by: Reza Yazdani <reyazda@microsoft.com>