main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\nn\optimizer\fused_adam.py
20 5 -
A
colossalai\tensor\moe_tensor\api.py
11 5 -
A
colossalai\nn\optimizer\fused_sgd.py
20 4 -
A
applications\ColossalChat\coati\models\utils.py
18 4 -
A
colossalai\device\__init__.py
 4 -
A
colossalai\tensor\moe_tensor\moe_info.py
2 3 -
A
colossalai\nn\optimizer\lars.py
19 4 -
A
colossalai\device\alpha_beta_profiler.py
36 6 -
A
colossalai\pipeline\__init__.py
 2 -
A
applications\Colossal-LLaMA\inference\stream_chat_example.py
5 1 -
A
examples\community\roberta\pretraining\utils\WandbLog.py
12 2 -
A
examples\community\roberta\pretraining\run_pretraining.py
28 5 -
A
colossalai\zero\gemini\chunk\utils.py
6 4 -
A
colossalai\utils\__init__.py
 2 -
A
colossalai\tensor\d_tensor\__init__.py
 5 -
A
colossalai\shardformer\modeling\mistral.py
115 3 -
A
colossalai\shardformer\examples\performance_benchmark.py
7 4 -
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -