main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_infer\test_kernels\triton\test_context_attn_unpad.py
12 2 -
A
colossalai\legacy\nn\layer\utils\common.py
12 3 -
A
colossalai\fx\passes\shard_1d_pass.py
12 9 -
A
colossalai\fx\profiler\experimental\profiler_module\convolution.py
12 3 -
A
examples\community\roberta\pretraining\utils\WandbLog.py
12 2 -
A
applications\ColossalQA\colossalqa\retrieval_conversation_universal.py
12 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_bmm_handler.py
11 3 -
A
tests\test_fx\test_tracer\test_torchrec_model\test_dlrm_model.py
11 3 -
A
tests\test_booster\test_plugin\test_torch_fsdp_plugin.py
11 6 -
A
applications\ColossalChat\coati\utils\accumulative_meter.py
11 1 -
A
colossalai\inference\modeling\backends\attention_backend.py
11 2 -
A
colossalai\initialize.py
11 9 -
A
colossalai\accelerator\api.py
11 2 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\upscaling.py
11 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\fp16.py
11 3 -
A
tests\test_checkpoint_io\test_plugins_huggingface_compatibility.py
11 7 -
A
examples\tutorial\sequence_parallel\data\datasets\builder.py
11 2 -
A
colossalai\legacy\context\random\seed_manager.py
11 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_compatibility_with_gemini.py
11 5 -
A
colossalai\nn\init.py
11 3 -