GitLab

* polish sharded model docstr

* polish sharded optim docstr

* polish zero docstr

* polish shard strategy docstr

* polish code

* shard strategy receive pg in shard() / gather()

* update zero engine

* polish code

* Revert "polish code"

This reverts commit 8cf7ff08.

* Revert "rename variables"

This reverts commit e99af94a.

* Revert "remove surplus imports"

This reverts commit 46add4a5.

* Revert "update sharded optim and fix zero init ctx"

This reverts commit 57567ee7.

Co-authored-by: github-actions <github-actions@github.com>

fixed mem monitor device

* better logger using rich

* remove deepspeed in zero requirements