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
ver217 26b7aac0be
[zero] reorganize zero/gemini folder structure (#3424)
2 years ago
..
layers [embedding] rename FreqAwareEmbedding -> CachedEmbedding (#1699) 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