You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/nn/parallel
digger yu 1878749753
[nfc] fix typo colossalai/nn (#3887)
2 years ago
..
layers [nfc] fix typo colossalai/nn (#3887) 2 years ago
__init__.py [zero] reorganize zero/gemini folder structure (#3424) 2 years ago
data_parallel.py [zero] reorganize zero/gemini folder structure (#3424) 2 years ago
reducer.py [ddp] ColoDDP uses bucket all-reduce (#1177) 2 years ago