ColossalAI/colossalai/engine/paramhooks
..
__init__.py
_param_hookmgr.py