main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalChat\coati\distributed\zero_bubble\consumer.py
44 1 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -
A
colossalai\legacy\nn\layer\parallel_2p5d\__init__.py
 2 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\output_generator.py
15 5 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit.py
2 3 -
A
colossalai\legacy\nn\layer\parallel_2p5d\_utils.py
3 3 -
A
tests\test_booster\test_accelerator.py
1 5 -
A
applications\ColossalChat\tests\test_lora.py
7 3 -
A
tests\test_fp8\test_fp8_cast.py
2 2 -
A
tests\test_fp8\test_fp8_allreduce.py
5 2 -
A
tests\test_cluster\test_device_mesh_manager.py
2 2 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
17 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\ray\detached_trainer_base.py
44 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -