main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\jit\option.py
10 5 -
A
colossalai\kernel\triton\rms_layernorm.py
10 1 -
A
colossalai\kernel\triton\softmax.py
10 2 -
A
colossalai\lazy\construction.py
10 1 -
A
tests\test_shardformer\test_model\test_shard_deepseek_v3.py
10 3 -
A
extensions\base_extension.py
10 3 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\conv.py
10 3 -
A
examples\images\diffusion\ldm\models\diffusion\dpm_solver\sampler.py
10 3 -
A
colossalai\legacy\engine\schedule\_non_pipeline_schedule.py
10 2 -
A
tests\test_analyzer\test_fx\test_nested_ckpt.py
10 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\normal_pooling_generator.py
10 5 -
A
tests\test_shardformer\test_model\test_shard_deepseek.py
10 5 -
A
tests\test_shardformer\test_layer\test_qkv_fused_linear_1d.py
10 3 -
A
tests\test_shardformer\test_layer\test_dropout.py
10 3 -
A
tests\test_optimizer\test_adam_optim.py
10 2 -
A
tests\test_moe\test_moe_ep_tp.py
10 13 -
A
applications\Colossal-LLaMA\colossal_llama\tokenizer\init_tokenizer.py
10 1 -
A
colossalai\legacy\trainer\hooks\_checkpoint_hook.py
10 4 -
A
tests\test_infer\test_async_engine\test_async_engine.py
10 1 -
A
applications\ColossalQA\colossalqa\chain\memory\summary.py
10 1 -