main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\booster\mixed_precision\__init__.py
2 3 -
A
tests\test_fx\test_tracer\test_torchrec_model\test_deepfm_model.py
10 3 -
A
colossalai\legacy\pipeline\rpc\_pipeline_schedule.py
58 2 -
A
tests\test_fx\test_tracer\test_torchvision_model\test_torchvision_model.py
7 9 -
A
applications\ColossalChat\coati\quant\llama_gptq\model_utils.py
4 1 -
A
tests\test_infer\test_async_engine\test_async_engine.py
10 1 -
A
colossalai\legacy\pipeline\utils.py
59 2 -
A
tests\test_infer\test_batch_bucket.py
1 1 -
A
colossalai\booster\mixed_precision\fp16_apex.py
 3 -
A
tests\test_infer\test_models\test_baichuan.py
 2 -
A
tests\test_infer\test_models\test_attention.py
4 1 -
A
tests\test_infer\test_kernels\cuda\test_rms_layernorm.py
1 1 -
A
tests\test_infer\test_kernels\cuda\test_kv_cache_memcpy.py
9 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 -
A
colossalai\fx\tracer\meta_patch\__init__.py
 3 -
A
colossalai\testing\random.py
3 2 -