main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\triton\__init__.py
 5 -
A
colossalai\shardformer\layer\normalization.py
23 5 -
A
colossalai\kernel\triton\context_attention.py
17 5 -
A
colossalai\booster\plugin\low_level_zero_plugin.py
43 5 -
A
tests\test_zero\test_gemini\test_grad_accum.py
18 4 -
A
tests\test_zero\test_gemini\test_fwd_bwd.py
13 4 -
A
colossalai\pipeline\p2p.py
126 4 4
A
colossalai\nn\optimizer\hybrid_adam.py
17 4 -
A
tests\test_moe\test_moe_ep_tp.py
31 4 -
A
tests\test_checkpoint_io\test_gemini_checkpoint_io.py
20 4 -
A
examples\language\llama2\performance_evaluator.py
17 4 -
A
colossalai\inference\quant\smoothquant\models\llama.py
94 4 3
A
colossalai\shardformer\shard\sharder.py
39 4 -
A
colossalai\inference\quant\smoothquant\models\base_model.py
73 4 1
A
colossalai\shardformer\shard\shard_config.py
10 4 -
A
colossalai\inference\engine\engine.py
24 4 1
A
applications\ColossalQA\examples\webui_demo\server.py
 3 1
A
colossalai\shardformer\layer\_operation.py
81 3 1
A
colossalai\pipeline\schedule\one_f_one_b.py
52 3 1
A
colossalai\shardformer\layer\utils.py
36 3 -