main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\zero\gemini\memory_tracer\runtime_mem_tracer.py
19 5 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
A
examples\images\diffusion\ldm\modules\distributions\distributions.py
19 2 -
A
applications\ColossalMoE\train.py
19 5 -
A
colossalai\interface\optimizer.py
19 3 -
A
examples\language\bert\data.py
19 4 -
A
tests\test_shardformer\test_flash_attention.py
19 3 -
A
colossalai\nn\optimizer\lars.py
19 4 -
A
examples\images\diffusion\ldm\lr_scheduler.py
19 2 -
A
examples\images\diffusion\ldm\modules\ema.py
19 3 -
A
tests\test_pipeline\test_schedule\test_oneF_oneB.py
19 8 -
A
colossalai\fx\profiler\tensor.py
19 11 -
A
tests\test_fx\test_codegen\test_nested_activation_checkpoint_codegen.py
18 4 -
A
tests\test_fx\test_codegen\test_activation_checkpoint_codegen.py
18 4 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
18 5 -
A
colossalai\legacy\moe\manager.py
18 1 -
A
tests\test_legacy\test_tensor\common_utils\_utils.py
18 2 -
A
applications\ColossalChat\examples\inference\web_chatbot\utils.py
18 1 -
A
colossalai\nn\optimizer\cpu_adam.py
18 2 -
A
colossalai\legacy\inference\dynamic_batching\req_queue.py
18 1 -