main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\autochunk\estimate_memory.py
75 8 -
A
tests\test_fx\test_codegen\test_nested_activation_checkpoint_codegen.py
18 4 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\copyer.py
5 2 -
A
tests\test_fx\test_coloproxy.py
3 8 -
A
applications\ColossalEval\colossal_eval\models\base.py
6 1 -
A
tests\test_fx\test_graph_manipulation.py
5 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding.py
17 3 -
A
tests\test_fx\test_meta\test_backward.py
4 6 -
A
colossalai\autochunk\search_chunk.py
54 9 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_albert.py
3 3 -
A
tests\test_fx\test_tracer\test_hf_model\hf_tracer_utils.py
8 5 -
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
.github\workflows\doc_test_on_pr.yml
 5 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\utils\ckpt_io.py
7 1 -
A
applications\ColossalQA\tests\test_memory.py
7 1 -
A
colossalai\fx\tracer\meta_patch\__init__.py
 3 -
A
colossalai\testing\random.py
3 2 -