main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\booster\mixed_precision\bf16.py
 1 -
A
colossalai\booster\mixed_precision\fp16_apex.py
 3 -
A
colossalai\booster\mixed_precision\fp16_naive.py
1 3 -
A
colossalai\booster\mixed_precision\fp16_torch.py
14 2 -
A
colossalai\booster\mixed_precision\fp8.py
 1 -
A
colossalai\booster\mixed_precision\mixed_precision_base.py
1 4 -
A
colossalai\booster\plugin\__init__.py
 3 -
A
colossalai\booster\plugin\dp_plugin_base.py
3 3 -
A
colossalai\booster\plugin\gemini_plugin.py
68 11 -
A
colossalai\booster\plugin\hybrid_parallel_plugin.py
162 13 -
A
colossalai\booster\plugin\low_level_zero_plugin.py
103 20 -
A
colossalai\booster\plugin\moe_hybrid_parallel_plugin.py
44 31 -
A
colossalai\booster\plugin\plugin_base.py
12 3 -
A
colossalai\booster\plugin\pp_plugin_base.py
1 3 -
A
colossalai\booster\plugin\torch_ddp_plugin.py
40 11 -
A
colossalai\booster\plugin\torch_fsdp_plugin.py
68 10 -
A
colossalai\checkpoint_io\__init__.py
 4 -
A
colossalai\checkpoint_io\checkpoint_io_base.py
32 8 -
A
colossalai\checkpoint_io\general_checkpoint_io.py
31 14 -
A
colossalai\checkpoint_io\hybrid_parallel_checkpoint_io.py
181 17 -