main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\batch_norm_generator.py
31 12 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding_tablewise.py
30 3 -
A
colossalai\inference\modeling\layers\attention.py
30 1 -
A
examples\images\diffusion\ldm\modules\midas\midas\transforms.py
30 2 -
A
examples\community\fp8\mnist\main.py
30 2 -
A
colossalai\legacy\utils\profiler\legacy\prof_utils.py
30 2 -
A
examples\images\diffusion\scripts\sample_diffusion.py
30 4 -
A
examples\community\roberta\pretraining\nvidia_bert_dataset_provider.py
30 3 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_amp_optimizer.py
30 4 -
A
tests\test_auto_parallel\test_ckpt_solvers\test_linearize.py
30 3 -
A
applications\ColossalChat\coati\experience_maker\naive.py
30 3 -
A
colossalai\shardformer\policies\deepseek_v3.py
30 2 -
A
colossalai\fx\profiler\opcount.py
29 14 -
A
colossalai\nn\optimizer\fused_lamb.py
29 4 -
A
colossalai\auto_parallel\checkpoint\ckpt_solver_base.py
29 4 -
A
tests\test_booster\test_plugin\test_3d_plugin.py
29 3 -
A
tests\test_autochunk\test_autochunk_alphafold\benchmark_autochunk_alphafold.py
29 5 -
A
examples\images\diffusion\ldm\modules\midas\midas\blocks.py
29 2 -
A
examples\images\diffusion\ldm\data\teyvat.py
29 3 -
A
applications\ColossalEval\colossal_eval\models\chatglm.py
29 3 -