main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\trainer\hooks\_checkpoint_hook.py
10 4 -
A
tests\test_fx\test_tracer\test_bias_addition_module.py
10 4 -
A
colossalai\legacy\zero\gemini\paramhooks\_param_hookmgr.py
10 2 -
A
extensions\base_extension.py
10 3 -
A
examples\images\diffusion\ldm\models\diffusion\dpm_solver\sampler.py
10 3 -
A
applications\ColossalQA\colossalqa\chain\memory\summary.py
10 1 -
A
colossalai\fx\profiler\dataflow.py
10 9 -
A
applications\ColossalChat\coati\trainer\callbacks\base.py
10 1 -
A
colossalai\fx\profiler\experimental\profiler_function\normalization.py
10 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\fx\profiler\shard_utils.py
11 4 -
A
colossalai\tensor\moe_tensor\api.py
11 5 -
A
tests\test_booster\test_plugin\test_torch_fsdp_plugin.py
11 6 -
A
tests\test_legacy\test_layers\test_2d\checks_2d\check_layer_2d.py
11 4 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\upscaling.py
11 2 -
A
colossalai\inference\modeling\backends\attention_backend.py
11 2 -
A
applications\ColossalChat\coati\utils\accumulative_meter.py
11 1 -
A
colossalai\initialize.py
11 9 -
A
colossalai\accelerator\api.py
11 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_compatibility_with_gemini.py
11 5 -