main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\fx\passes\experimental\adding_shape_consistency_pass.py
23 1 -
A
colossalai\nn\layer\colossalai_layer\_utils.py
10 1 -
A
colossalai\amp\torch_amp\__init__.py
3 1 -
A
colossalai\nn\layer\colossalai_layer\embedding.py
 1 -
A
colossalai\auto_parallel\solver\strategy\unary_elementwise_generator.py
14 1 -
A
colossalai\nn\layer\colossalai_layer\normalization.py
2 1 -
A
colossalai\nn\layer\parallel_1d\layers.py
 4 -
A
colossalai\nn\layer\moe\_operation.py
29 1 -
A
colossalai\builder\__init__.py
 2 -
A
colossalai\nn\layer\moe\layers.py
22 3 -
A
colossalai\nn\layer\wrapper\__init__.py
 1 -
A
colossalai\nn\loss\loss_1d.py
6 1 -
A
colossalai\nn\loss\__init__.py
7 1 -
A
colossalai\nn\layer\parallel_3d\__init__.py
 1 -
A
colossalai\nn\layer\parallel_2p5d\_operation.py
84 1 -
A
.github\ISSUE_TEMPLATE\documentation.yml
 1 -
A
.github\workflows\assign_reviewer.yml
 1 -
A
colossalai\auto_parallel\solver\op_handler\strategy_generator.py
3 2 -
A
colossalai\engine\gradient_handler\_base_gradient_handler.py
2 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\feed_forward.h
7 1 -