main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\gptj.py
71 4 -
A
colossalai\shardformer\policies\bloom.py
70 4 -
A
examples\images\dreambooth\train_dreambooth_colossalai.py
70 5 -
A
colossalai\legacy\context\parallel_context.py
69 3 -
A
colossalai\shardformer\policies\opt.py
69 3 -
A
colossalai\shardformer\modeling\vit.py
69 2 -
A
colossalai\booster\plugin\gemini_plugin.py
68 11 -
A
colossalai\booster\plugin\torch_fsdp_plugin.py
68 10 -
A
applications\ColossalChat\coati\models\generation.py
68 2 -
A
tests\test_pipeline\test_schedule\test_zerobubble_pp.py
68 2 -
A
colossalai\autochunk\autochunk_codegen.py
67 3 -
A
examples\images\diffusion\ldm\models\diffusion\ddim.py
67 3 -
A
applications\ColossalChat\coati\quant\llama_gptq\quant.py
66 1 -
A
applications\ColossalChat\coati\ray\experience_maker_holder.py
66 1 -
A
extensions\csrc\kernel\x86\cpu_adam.cpp
66 1 -
A
colossalai\moe\_operation.py
66 14 -
A
applications\ColossalChat\examples\training_scripts\train_ppo.py
65 4 1
A
applications\ColossalEval\colossal_eval\models\vllm.py
63 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\matmul_handler.py
62 9 -
A
colossalai\shardformer\modeling\deepseek_v3.py
62 2 -