main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\amp\naive_amp\mixed_precision_mixin\fp16.py
11 3 -
A
examples\images\vit\vit_train_demo.py
35 3 -
A
colossalai\kernel\triton\__init__.py
 11 -
A
examples\inference\benchmark_ops\benchmark_decoding_attn.py
4 1 -
A
applications\ColossalChat\examples\inference\web_chatbot\requirements.txt
 1 -
A
examples\inference\benchmark_ops\benchmark_fused_rotary_embdding_unpad.py
7 1 -
A
colossalai\kernel\triton\flash_decoding.py
18 3 -
A
examples\inference\benchmark_ops\benchmark_rmsnorm.py
8 1 -
A
colossalai\auto_parallel\checkpoint\__init__.py
 2 -
A
examples\language\bert\benchmark_utils.py
20 2 -
A
examples\language\bert\benchmark.sh
 1 -
A
examples\inference\llama\llama_generation.py
2 3 -
A
examples\inference\llama\benchmark_llama3.py
20 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\utils\__init__.py
 2 -