ColossalAI/colossalai/nn/layer/parallel_3d
LuGY fb5bc6cb28 [NFC] polish colossalai/nn/layer/parallel_3d/layers.py code style (#966) 2022-05-17 10:25:06 +08:00
..
__init__.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
_operation.py [TP] allow layernorm without bias (#750) 2022-04-14 11:43:56 +08:00
_utils.py moved env variables to global variables; (#215) 2022-02-15 11:31:13 +08:00
layers.py [NFC] polish colossalai/nn/layer/parallel_3d/layers.py code style (#966) 2022-05-17 10:25:06 +08:00