main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalChat\examples\training_scripts\train_sft.py
40 11 2
A
colossalai\booster\plugin\gemini_plugin.py
68 11 -
A
colossalai\booster\plugin\torch_ddp_plugin.py
40 11 -
A
colossalai\fx\profiler\__init__.py
 11 -
A
colossalai\shardformer\policies\base_policy.py
21 11 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\linear.py
26 12 -
A
colossalai\auto_parallel\tensor_shard\node_handler\linear_handler.py
18 12 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\batch_norm_generator.py
31 12 -
A
colossalai\auto_parallel\tensor_shard\solver\strategies_constructor.py
29 12 -
A
colossalai\autochunk\trace_flow.py
115 12 -
A
tests\test_zero\test_gemini\test_optim.py
22 12 -
B
colossalai\shardformer\modeling\deepseek.py
80 12 3
A
colossalai\shardformer\policies\deepseek.py
61 12 -
A
colossalai\auto_parallel\passes\runtime_apply_pass.py
53 13 -
A
colossalai\auto_parallel\tensor_shard\sharding_strategy.py
32 13 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_bert.py
3 13 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_opt.py
2 13 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_t5.py
3 13 -
A
tests\test_moe\test_moe_ep_tp.py
10 13 -
A
colossalai\booster\plugin\hybrid_parallel_plugin.py
162 13 3