main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\device\__init__.py
 4 -
A
tests\test_shardformer\test_with_torch_ddp.py
9 4 -
A
colossalai\nn\layer\layernorm.py
8 1 -
A
tests\test_smoothquant\test_llama_mlp.py
2 1 -
A
applications\ColossalChat\coati\ray\lora_constructor.py
20 1 -
A
tests\test_smoothquant\test_sq_rotary_embedding.py
2 1 -
A
colossalai\nn\layer\utils.py
1 2 -
A
tests\test_tensor\test_dtensor\test_comm_spec.py
16 4 -
A
colossalai\device\calc_pipeline_strategy.py
28 2 -
A
tests\test_zero\test_low_level\test_zero1_2.py
25 3 -
A
tests\test_zero\test_low_level\test_mem_leak.py
6 1 -
A
tests\test_zero\test_gemini\test_chunkv2.py
14 6 -
A
tests\test_zero\test_gemini\test_chunk_mgrv2.py
8 4 -
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 -