ColossalAI/colossalai/nn/layer/colossalai_layer
zbian 653b0a620e added skip_bias_add for non-tp linear 2022-11-09 15:41:08 +08:00
..
__init__.py
_utils.py
dropout.py
embedding.py
linear.py
normalization.py