GitLab

* [zero] refactor low-level zero folder structure

* [zero] fix legacy zero import path

* [zero] fix legacy zero import path

* [zero] remove useless import

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor legacy zero import path

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor gemini folder structure

* [zero] refactor legacy zero import path

* [zero] fix test import path

* [zero] fix test

* [zero] fix circular import

* [zero] update import

* remove forced FP32 modules

* correct no_shard-contexts' positions

* [test] refactored with the new rerun decorator

* polish test case

* add tensor placement policies

* polish comments

* polish comments

* update moe unit tests

* adapt post grad hooks for not-shard parameters
* adapt optimizer for not-shard parameters
* offload gradients for not-replicated parameters

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

* remove hybrid adam in test_moe_zero_optim

* fix activation checkpointing and its unitest

* add cpu adam counter for all cpu adam

* fixed updating error in adam kernel