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
Geng Zhang 0e06f62160
[NFC] polish colossalai/nn/layer/parallel_sequence/_operation.py code style (#1266)
2 years ago
..
colossalai_layer
moe
parallel_1d [Tensor] 1d row embedding (#1075) 3 years ago
parallel_2d [NFC] polish colossalai/nn/layer/parallel_2d/layers.py code style (#976) 3 years ago
parallel_2p5d
parallel_3d [NFC] polish colossalai/nn/layer/parallel_3d/_operation.py code style (#1258) 2 years ago
parallel_sequence [NFC] polish colossalai/nn/layer/parallel_sequence/_operation.py code style (#1266) 2 years ago
utils [NFC] polish colossalai/nn/layer/utils/common.py code style (#983) 3 years ago
vanilla
wrapper [pipeline] refactor the pipeline module (#1087) 2 years ago
__init__.py
base_layer.py