ColossalAI/colossalai/nn
Jiarui Fang 986f8cbaa7
[inference] overlap comm and compute in Linear1D_Row when stream_chunk_num > 1 (#1876)
2022-11-10 17:36:42 +08:00
..
_ops
graph
layer [inference] overlap comm and compute in Linear1D_Row when stream_chunk_num > 1 (#1876) 2022-11-10 17:36:42 +08:00
loss
lr_scheduler
metric
optimizer
parallel
__init__.py
init.py