main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\meta_profiler\meta_registry\norm.py
2 8 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\pooling.py
4 8 -
A
tests\test_fx\test_tracer\test_patched_module.py
27 8 -
A
colossalai\checkpoint_io\moe_checkpoint.py
155 8 -
A
tests\test_fx\test_pipeline\test_timm_model\test_timm.py
4 7 -
A
tests\test_fx\test_pipeline\test_hf_model\test_t5.py
3 7 -
A
colossalai\inference\core\engine.py
14 7 -
A
tests\test_fx\test_pipeline\test_hf_model\test_opt.py
2 7 -
A
colossalai\zero\gemini\placement_policy.py
58 7 -
A
colossalai\zero\gemini\chunk\search_utils.py
25 7 -
A
colossalai\fx\tracer\meta_patch\patched_function\torch_ops.py
47 7 -
A
tests\test_checkpoint_io\test_gemini_torch_compability.py
20 7 -
A
colossalai\testing\__init__.py
 7 -
A
colossalai\auto_parallel\tensor_shard\utils\broadcast.py
23 7 -
A
tests\test_checkpoint_io\test_plugins_huggingface_compatibility.py
11 7 -
A
colossalai\auto_parallel\tensor_shard\solver\cost_graph.py
49 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\unary_elementwise_handler.py
2 7 -
A
colossalai\fx\tracer\_tracer_utils.py
12 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\reshape_generator.py
45 7 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -