ColossalAI/colossalai/nn/layer/colossalai_layer
zbian 61e687831d fixed using zero with tp cannot access weight correctly 2023-02-28 10:52:30 +08:00
..
__init__.py
_utils.py fixed using zero with tp cannot access weight correctly 2023-02-28 10:52:30 +08:00
dropout.py fixed using zero with tp cannot access weight correctly 2023-02-28 10:52:30 +08:00
embedding.py
linear.py
normalization.py