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 [NFC] polish colossalai/nn/layer/colossalai_layer/dropout.py code style (#1568) 2022-09-08 22:11:04 +08:00
embedding.py
linear.py added skip_bias_add for non-tp linear 2022-11-09 15:41:08 +08:00
normalization.py [TP] allow layernorm without bias (#750) 2022-04-14 11:43:56 +08:00