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
アマデウス 0fedef4f3c
Layer integration (#83)
3 years ago
..
parallel_1d Layer integration (#83) 3 years ago
parallel_2d Layer integration (#83) 3 years ago
parallel_2p5d Layer integration (#83) 3 years ago
parallel_3d Layer integration (#83) 3 years ago
parallel_sequence Migrated project 3 years ago
vanilla Layer integration (#83) 3 years ago
wrapper Migrated project 3 years ago
__init__.py Layer integration (#83) 3 years ago
_common_utils.py Layer integration (#83) 3 years ago
base_layer.py Migrated project 3 years ago
colossalai_layer.py Layer integration (#83) 3 years ago
fused_bias_gelu.py Develop/experiments (#59) 3 years ago