ColossalAI/colossalai/nn/layer
Frank Lee 35813ed3c4
update examples and sphnix docs for the new api (#63)
2021-12-13 22:07:01 +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 Develop/experiments (#59) 2021-12-09 15:08:29 +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