main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\pipeline\stage_manager.py
25 3 -
A
tests\kit\model_zoo\transformers\gpt.py
6 3 -
A
examples\inference\run_benchmark.sh
 3 -
A
colossalai\shardformer\policies\chatglm2.py
32 3 -
A
colossalai\shardformer\layer\utils.py
36 3 -
A
colossalai\pipeline\schedule\one_f_one_b.py
52 3 1
A
colossalai\pipeline\schedule\interleaved_pp.py
57 3 2
A
colossalai\nn\layer\__init__.py
 2 -
A
colossalai\moe\routers.py
33 2 1
A
colossalai\nn\lr_scheduler\delayed.py
45 2 -
A
colossalai\moe\manager.py
18 2 -
A
colossalai\moe\experts.py
32 2 1
A
colossalai\moe\__init__.py
 2 -
A
colossalai\legacy\zero\gemini\stateful_tensor_mgr.py
18 2 -
A
colossalai\legacy\utils\activation_checkpoint.py
42 2 1
A
colossalai\moe\checkpoint.py
120 2 1
A
colossalai\legacy\zero\gemini\__init__.py
 2 -
A
colossalai\legacy\tensor\process_group.py
53 2 1
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 2 -
A
colossalai\legacy\nn\layer\parallel_2p5d\layers.py
 2 -