main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\images\vit\vit_benchmark.py
21 6 -
A
examples\images\diffusion\setup.py
 2 -
A
colossalai\initialize.py
11 9 -
A
examples\images\diffusion\train_colossalai.sh
 5 -
A
colossalai\amp\naive_amp\grad_scaler\dynamic_grad_scaler.py
16 4 -
A
examples\images\dreambooth\colossalai.sh
 3 -
A
colossalai\interface\model.py
46 3 -
A
examples\images\dreambooth\dreambooth.sh
 2 -
A
applications\ColossalChat\benchmarks\dummy_dataset.py
8 2 -
A
examples\inference\benchmark_ops\benchmark_flash_decoding_attention.py
6 1 -
A
examples\inference\benchmark_ops\benchmark_decoding_attn.py
4 1 -
A
examples\images\resnet\test_ci.sh
 1 -
A
examples\images\resnet\eval.py
 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 -