main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\layer\utils.py
76 5 -
A
colossalai\_analyzer\fx\passes\shape_prop.py
42 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\normal_pooling_generator.py
10 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\output_generator.py
15 5 -
A
tests\test_analyzer\test_fx\test_bias_addition.py
18 5 -
A
examples\tutorial\sequence_parallel\model\bert.py
39 5 -
A
examples\tutorial\sequence_parallel\config.py
 5 -
A
examples\tutorial\sequence_parallel\train.py
36 5 -
A
.github\workflows\scripts\generate_release_draft.py
15 5 -
A
examples\tutorial\new_api\glue_bert\finetune.py
24 5 -
A
examples\tutorial\new_api\cifar_vit\train.py
21 5 -
A
examples\tutorial\hybrid_parallel\config.py
 5 -
A
tests\test_analyzer\test_subclasses\test_flop_tensor.py
3 5 -
A
examples\language\palm\train.py
16 5 -
A
examples\tutorial\new_api\cifar_resnet\train.py
20 5 -
A
examples\tutorial\hybrid_parallel\train.py
17 5 -
A
examples\language\performance_evaluator.py
20 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\registry.py
4 5 -
A
colossalai\nn\optimizer\fused_adam.py
20 5 -
A
examples\language\grok-1\inference_tp.py
 5 -