main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\booster\plugin\gemini_plugin.py
47 9 -
B
colossalai\booster\plugin\hybrid_parallel_plugin.py
160 8 4
B
colossalai\zero\gemini\gemini_ddp.py
173 8 4
A
colossalai\inference\__init__.py
 7 -
A
colossalai\zero\gemini\chunk\chunk.py
112 7 -
A
colossalai\pipeline\schedule\generate.py
74 6 1
A
colossalai\nn\optimizer\cpu_adam.py
15 6 -
A
colossalai\zero\gemini\chunk\manager.py
52 6 -
A
colossalai\zero\gemini\gemini_optimizer.py
144 6 2
A
colossalai\shardformer\policies\auto_policy.py
5 6 -
A
tests\test_booster\test_plugin\test_gemini_plugin.py
25 6 -
A
examples\language\llama2\benchmark.py
18 6 1
A
tests\test_zero\test_gemini\test_optim.py
22 6 -
B
colossalai\zero\low_level\low_level_optim.py
187 6 5
A
colossalai\shardformer\modeling\llama.py
77 6 3
A
applications\ColossalEval\colossal_eval\dataset\__init__.py
 5 -
A
colossalai\booster\plugin\low_level_zero_plugin.py
43 5 -
A
colossalai\shardformer\layer\normalization.py
23 5 -
A
colossalai\kernel\triton\context_attention.py
17 5 -
A
colossalai\kernel\triton\__init__.py
 5 -