main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\nn\optimizer\lamb.py
17 4 -
A
applications\Colossal-LLaMA\dataset\prepare_sft_dataset.py
18 1 -
A
colossalai\nn\optimizer\nvme_optimizer.py
44 4 -
A
colossalai\fx\passes\experimental\adding_shape_consistency_pass.py
23 3 -
A
colossalai\fx\passes\meta_info_prop.py
18 23 -
A
applications\ColossalChat\coati\trainer\callbacks\__init__.py
 1 -
A
colossalai\pipeline\schedule\_utils.py
36 3 -
A
colossalai\fx\passes\passes_for_gpt2_test.py
50 6 -
A
colossalai\pipeline\schedule\generate.py
74 9 -
A
applications\Colossal-LLaMA\inference\inference_example.py
 1 -
A
colossalai\quantization\fp8.py
110 6 -
A
colossalai\quantization\bnb_config.py
23 1 -
A
colossalai\pipeline\weight_grad_store.py
8 2 -
A
colossalai\pipeline\stage_manager.py
33 2 -
A
.github\ISSUE_TEMPLATE\bug-report.yml
 2 -
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 -