main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\checkpoint\ckpt_solver_rotor.py
85 6 -
A
colossalai\shardformer\modeling\gptj.py
85 7 -
A
colossalai\legacy\nn\layer\parallel_2p5d\_operation.py
84 4 -
A
colossalai\auto_parallel\offload\region_manager.py
82 2 -
A
colossalai\inference\batch_bucket.py
81 3 -
A
colossalai\legacy\nn\layer\parallel_2d\_operation.py
81 4 -
A
colossalai\legacy\zero\sharded_optim\sharded_optim_v2.py
80 2 -
A
colossalai\shardformer\modeling\deepseek.py
80 12 -
A
colossalai\autochunk\utils.py
79 10 -
A
colossalai\legacy\amp\torch_amp\_grad_scaler.py
79 2 -
A
examples\images\diffusion\ldm\data\imagenet.py
79 2 3
A
examples\images\dreambooth\train_dreambooth.py
78 2 1
A
colossalai\pipeline\schedule\one_f_one_b.py
78 17 -
A
colossalai\inference\kv_cache\kvcache_manager.py
78 4 -
A
colossalai\inference\modeling\layers\distrifusion.py
78 1 -
A
colossalai\shardformer\modeling\mixtral.py
78 16 -
A
tests\test_shardformer\test_model\_utils.py
77 4 -
A
examples\tutorial\opt\opt\run_clm.py
77 6 -
A
examples\images\diffusion\ldm\modules\image_degradation\bsrgan_light.py
77 3 -
A
applications\ColossalEval\colossal_eval\evaluate\dataset_evaluator\dataset_evaluator.py
77 6 -