main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\zero\gemini\chunk\manager.py
54 15 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\matmul_strategy_generator.py
87 15 -
A
colossalai\zero\gemini\chunk\chunk.py
114 15 -
A
colossalai\fx\proxy.py
24 14 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\utils.py
26 14 -
A
colossalai\checkpoint_io\general_checkpoint_io.py
31 14 -
A
colossalai\shardformer\policies\mixtral.py
63 14 -
A
colossalai\moe\_operation.py
66 14 -
A
tests\test_shardformer\test_model\test_shard_mixtral.py
17 14 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\strategy_generator.py
33 14 -
A
colossalai\fx\profiler\opcount.py
29 14 -
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
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_bert.py
3 13 -
A
colossalai\auto_parallel\tensor_shard\sharding_strategy.py
32 13 -
A
colossalai\fx\passes\adding_split_node_pass.py
90 13 -
A
colossalai\auto_parallel\passes\runtime_apply_pass.py
53 13 -
A
colossalai\shardformer\modeling\opt.py
97 13 -
A
colossalai\auto_parallel\tensor_shard\solver\strategies_constructor.py
29 12 -