ColossalAI/colossalai/nn/parallel
..
layers
__init__.py
data_parallel.py
reducer.py