GitLab

* adapt model weight initialization for methods in Pytorch nn.init

* support existing sharded and unsharded parameters in zero

* add unitest for moe-zero model init

* polish moe gradient handler