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/layer
HELSON 632e622de8
overlap computation and communication in 2d operations (#75)
3 years ago
..
non_parallel_layers update examples and sphnix docs for the new api (#63) 3 years ago
parallel_1d Develop/experiments (#59) 3 years ago
parallel_2d overlap computation and communication in 2d operations (#75) 3 years ago
parallel_2p5d Develop/experiments (#59) 3 years ago
parallel_3d Develop/experiments (#59) 3 years ago
parallel_sequence Migrated project 3 years ago
wrapper Migrated project 3 years ago
__init__.py Develop/experiments (#59) 3 years ago
_common_utils.py Develop/experiments (#59) 3 years ago
_parallel_utilities.py Migrated project 3 years ago
base_layer.py Migrated project 3 years ago
fused_bias_gelu.py Develop/experiments (#59) 3 years ago