ColossalAI/colossalai/nn/layer
Jiarui Fang c2947dadf1
[inference] streaming Linear 1D Row inference (#1874)
2022-11-10 17:03:21 +08:00
..
colossalai_layer added skip_bias_add for non-tp linear 2022-11-09 15:41:08 +08:00
moe
parallel_1d [inference] streaming Linear 1D Row inference (#1874) 2022-11-10 17:03:21 +08:00
parallel_2d
parallel_2p5d
parallel_3d
parallel_sequence
utils
vanilla added skip_bias_add for non-tp linear 2022-11-09 15:41:08 +08:00
wrapper
__init__.py
base_layer.py