ColossalAI/colossalai/legacy/engine/gradient_accumulation
..
__init__.py
_gradient_accumulation.py