ColossalAI/colossalai/legacy/zero/gemini/paramhooks
..
__init__.py
_param_hookmgr.py