main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\nn\parallel\layers\colo_module.py
12 3 -
A
examples\inference\stable_diffusion\compute_metric.py
12 1 -
A
colossalai\booster\plugin\plugin_base.py
12 3 -
A
colossalai\legacy\nn\layer\utils\common.py
12 3 -
A
tests\kit\model_zoo\registry.py
12 8 -
A
colossalai\fx\passes\shard_1d_pass.py
12 9 -
A
tests\test_infer\test_streamingllm.py
12 1 -
A
examples\community\roberta\pretraining\utils\WandbLog.py
12 2 -
A
tests\test_moe\moe_utils.py
12 10 -
A
colossalai\fx\profiler\experimental\profiler_module\convolution.py
12 3 1
A
colossalai\fx\profiler\shard_utils.py
11 4 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\upscaling.py
11 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_compatibility_with_gemini.py
11 5 -
A
tests\test_checkpoint_io\test_plugins_huggingface_compatibility.py
11 7 -
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
colossalai\amp\naive_amp\mixed_precision_mixin\fp16.py
11 3 -
A
colossalai\legacy\context\random\seed_manager.py
11 2 -