main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalEval\examples\dataset_evaluation\inference.py
37 9 2
A
colossalai\pipeline\schedule\generate.py
74 9 -
B
colossalai\shardformer\layer\_operation.py
189 9 3
A
colossalai\fx\passes\shard_1d_pass.py
12 9 -
A
colossalai\fx\profiler\dataflow.py
10 9 -
A
colossalai\tensor\utils.py
33 10 -
A
colossalai\zero\low_level\bookkeeping\gradient_store.py
18 10 -
A
examples\images\diffusion\configs\train_colossalai.yaml
 10 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\__init__.py
 10 -
A
colossalai\autochunk\utils.py
79 10 -
A
tests\test_moe\moe_utils.py
12 10 -
A
tests\test_moe\test_kernel.py
7 10 -
A
tests\test_zero\test_gemini\test_grad_clip.py
12 10 -
A
colossalai\booster\plugin\torch_fsdp_plugin.py
68 10 -
A
colossalai\shardformer\layer\dropout.py
8 10 -
A
colossalai\fx\profiler\tensor.py
19 11 -
A
examples\images\diffusion\main.py
90 11 -
A
colossalai\kernel\triton\__init__.py
 11 -
A
applications\ColossalChat\examples\training_scripts\train_dpo.py
40 11 3
A
colossalai\auto_parallel\tensor_shard\initialize.py
34 11 -