main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\pipeline\schedule\_utils.py
36 3 -
A
colossalai\cluster\process_group_mesh.py
36 17 -
A
applications\ColossalChat\coati\trainer\sft.py
36 6 -
A
applications\ColossalChat\examples\data_preparation_scripts\prepare_dataset.py
35 3 -
A
tests\test_infer\test_kernels\triton\kernel_utils.py
35 1 -
A
examples\images\vit\vit_train_demo.py
35 3 -
A
examples\images\diffusion\ldm\models\diffusion\plms.py
35 3 -
A
applications\ColossalChat\coati\dataset\utils.py
35 3 -
A
colossalai\legacy\pipeline\pipeline_process_group.py
35 2 -
A
colossalai\fx\passes\split_module.py
35 8 -
A
colossalai\legacy\moe\layer\layers.py
35 2 -
A
colossalai\legacy\zero\init_ctx\init_context.py
35 2 -
A
colossalai\legacy\communication\collective.py
35 3 -
A
examples\community\roberta\preprocessing\mask.cpp
34 1 -
A
examples\language\gpt\hybridparallelism\finetune.py
34 8 -
A
colossalai\testing\comparison.py
34 3 -
A
colossalai\nn\layer\scaled_softmax.py
34 2 -
A
applications\ColossalChat\coati\ray\callbacks\performance_evaluator.py
34 1 -
A
colossalai\inference\logit_processors.py
34 1 -
A
colossalai\auto_parallel\tensor_shard\initialize.py
34 11 -