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
botbw 62cdac6b7b
[chore] remove redundant test case, print string & reduce test tokens
4 months ago
..
_C
_analyzer
accelerator
amp
auto_parallel
autochunk
booster [moe] solve dp axis issue 4 months ago
checkpoint_io [moe] implement tp 4 months ago
cli
cluster Revert "[moe] implement submesh initialization" 4 months ago
context
device
fx
inference
interface
kernel
lazy
legacy [moe] full test for deepseek and mixtral (pp + sp to fix) 4 months ago
logging
moe [moe] remove ops 4 months ago
nn
pipeline
quantization
shardformer [chore] remove redundant test case, print string & reduce test tokens 4 months ago
tensor
testing
utils
zero [test] fix test: test_zero1_2 4 months ago
__init__.py
initialize.py