main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\device\device_mesh.py
31 3 -
A
colossalai\engine\__init__.py
 1 -
A
colossalai\engine\_base_engine.py
28 2 -
A
colossalai\engine\gradient_accumulation\__init__.py
4 1 -
A
colossalai\engine\gradient_accumulation\_gradient_accumulation.py
37 1 -
A
colossalai\engine\gradient_handler\__init__.py
 1 -
A
colossalai\engine\gradient_handler\_base_gradient_handler.py
2 1 -
A
colossalai\engine\gradient_handler\_data_parallel_gradient_handler.py
2 1 -
A
colossalai\engine\gradient_handler\_moe_gradient_handler.py
7 1 -
A
colossalai\engine\gradient_handler\_pipeline_parallel_gradient_handler.py
13 2 -
A
colossalai\engine\gradient_handler\_sequence_parallel_gradient_handler.py
2 1 -
A
colossalai\engine\gradient_handler\_zero_gradient_handler.py
1 1 -
A
colossalai\engine\gradient_handler\utils.py
8 1 -
A
colossalai\engine\schedule\__init__.py
 1 -
A
colossalai\engine\schedule\_base_schedule.py
37 1 -
A
colossalai\engine\schedule\_non_pipeline_schedule.py
10 1 -
A
colossalai\engine\schedule\_pipeline_schedule.py
155 2 2
A
colossalai\engine\schedule\_pipeline_schedule_v2.py
23 1 -
A
colossalai\fx\__init__.py
 4 -
A
colossalai\fx\codegen\__init__.py
 2 -