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
HELSON 1468e4bcfc
[zero] add constant placement policy (#1705)
2 years ago
..
layers [embedding] rename FreqAwareEmbedding -> CachedEmbedding (#1699) 2 years ago
__init__.py
data_parallel.py [zero] add constant placement policy (#1705) 2 years ago
reducer.py
utils.py