main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\language\performance_evaluator.py
20 5 -
A
colossalai\inference\core\plugin.py
20 2 -
A
colossalai\legacy\moe\openmoe\model\convert_openmoe_ckpt.py
19 1 -
A
tests\test_pipeline\test_schedule\test_interleaved.py
19 8 -
A
tests\test_optimizer\test_adam_kernel.py
19 5 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
A
examples\language\bert\data.py
19 4 -
A
colossalai\zero\gemini\memory_tracer\memory_monitor.py
19 4 -
A
tests\test_shardformer\test_flash_attention.py
19 3 -
A
examples\images\diffusion\ldm\lr_scheduler.py
19 2 -
A
tests\test_pipeline\test_schedule\test_oneF_oneB.py
19 8 -
A
colossalai\interface\optimizer.py
19 3 -
A
applications\ColossalMoE\train.py
19 5 -
A
colossalai\nn\optimizer\lars.py
19 4 -
A
examples\language\gpt\experiments\pipeline_parallel\train_gpt_pp.py
19 7 -
A
examples\images\diffusion\ldm\modules\distributions\distributions.py
19 2 -
A
examples\tutorial\new_api\glue_bert\data.py
19 3 -
A
colossalai\zero\gemini\memory_tracer\runtime_mem_tracer.py
19 5 -
A
colossalai\fx\profiler\tensor.py
19 11 -
A
examples\images\diffusion\ldm\modules\ema.py
19 3 -