main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\modeling\whisper.py
123 7 -
A
colossalai\shardformer\modeling\gpt2.py
123 21 -
A
colossalai\pipeline\schedule\v_schedule.py
122 1 1
A
colossalai\shardformer\layer\qkv_fused_linear.py
119 9 -
A
colossalai\checkpoint_io\utils.py
119 7 -
A
colossalai\autochunk\trace_flow.py
115 12 -
A
colossalai\legacy\utils\common.py
115 3 -
A
colossalai\legacy\zero\sharded_model\sharded_model_v2.py
114 3 -
A
colossalai\zero\gemini\chunk\chunk.py
114 15 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\model.py
111 4 -
A
examples\tutorial\sequence_parallel\data\tokenizer\bert_tokenization.py
111 2 -
A
examples\images\diffusion\ldm\models\diffusion\dpm_solver\dpm_solver.py
110 2 -
A
colossalai\quantization\fp8.py
110 6 -
A
colossalai\inference\core\llm_engine.py
108 2 -
A
examples\tutorial\sequence_parallel\data\datasets\indexed_dataset.py
107 3 -
A
colossalai\shardformer\layer\attn.py
107 9 -
A
colossalai\pipeline\schedule\interleaved_pp.py
106 15 -
A
examples\images\diffusion\ldm\modules\image_degradation\bsrgan.py
105 2 -
A
applications\ColossalChat\examples\community\ray\train_prompts_on_ray.py
104 1 -
A
colossalai\shardformer\modeling\bloom.py
104 12 -