ColossalAI/colossalai/nn/layer
HELSON 632e622de8
overlap computation and communication in 2d operations (#75)
2021-12-16 16:05:15 +08:00
..
non_parallel_layers update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00
parallel_1d Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
parallel_2d overlap computation and communication in 2d operations (#75) 2021-12-16 16:05:15 +08:00
parallel_2p5d Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
parallel_3d Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
parallel_sequence Migrated project 2021-10-28 18:21:23 +02:00
wrapper Migrated project 2021-10-28 18:21:23 +02:00
__init__.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
_common_utils.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
_parallel_utilities.py Migrated project 2021-10-28 18:21:23 +02:00
base_layer.py Migrated project 2021-10-28 18:21:23 +02:00
fused_bias_gelu.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00