main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_shardformer\test_layer\test_dist_crossentropy.py
2 5 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_zero_optimizer.py
22 5 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_naive_optimizer.py
25 4 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_amp_optimizer.py
30 4 -
A
tests\test_shardformer\test_flash_attention.py
19 3 -
A
tests\test_pipeline\test_stage_manager.py
8 4 -
A
tests\test_pipeline\test_schedule\test_zerobubble_pp.py
68 1 -
A
tests\test_pipeline\test_schedule\test_pipeline_schedule_utils.py
3 2 -
A
tests\test_pipeline\test_schedule\test_oneF_oneB.py
19 8 -
A
tests\test_pipeline\test_schedule\test_interleaved.py
19 8 -
A
tests\test_pipeline\test_pipeline_utils\test_whisper_pipeline_utils.py
8 2 -
A
tests\test_pipeline\test_pipeline_utils\test_t5_pipeline_utils.py
8 2 -
A
tests\test_pipeline\test_p2p_communication.py
17 8 -
A
tests\test_optimizer\test_nvme.py
7 5 -
A
tests\test_optimizer\test_lr_scheduler.py
2 1 -
A
tests\test_optimizer\test_dist_lamb.py
36 4 -
A
tests\test_optimizer\test_dist_galore.py
36 2 -
A
tests\test_optimizer\test_dist_came.py
47 4 -
A
tests\test_optimizer\test_dist_adafactor.py
50 4 -
A
tests\test_optimizer\test_adam_optim.py
17 3 -