main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\pipeline\schedule\v_schedule.py
122 1 1
A
colossalai\pipeline\schedule\zero_bubble_pp.py
128 2 -
A
colossalai\pipeline\stage_manager.py
33 2 -
A
colossalai\pipeline\weight_grad_store.py
8 2 -
A
colossalai\quantization\__init__.py
 1 -
A
colossalai\quantization\bnb.py
55 2 -
A
colossalai\quantization\bnb_config.py
23 1 -
A
colossalai\quantization\fp8.py
110 6 -
A
colossalai\quantization\fp8_config.py
 1 -
A
colossalai\quantization\fp8_hook.py
6 1 -
A
colossalai\quantization\utils.py
12 1 -
A
colossalai\shardformer\__init__.py
 2 -
A
colossalai\shardformer\_utils.py
15 3 -
A
colossalai\shardformer\examples\convergence_benchmark.py
15 4 -
A
colossalai\shardformer\examples\convergence_benchmark.sh
 2 -
A
colossalai\shardformer\examples\data.py
23 2 -
A
colossalai\shardformer\examples\performance_benchmark.py
7 4 -
A
colossalai\shardformer\layer\__init__.py
 5 -
A
colossalai\shardformer\layer\_operation.py
189 9 -
A
colossalai\shardformer\layer\attn.py
107 9 -