You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/nn
HELSON 8213f89fd2
[gemini] add fake_release_chunk for keep-gathered chunk in the inference mode (#2671)
2 years ago
..
_ops
layer
loss
lr_scheduler
metric
optimizer [doc] add CVPR tutorial (#2666) 2 years ago
parallel [gemini] add fake_release_chunk for keep-gathered chunk in the inference mode (#2671) 2 years ago
__init__.py
init.py