main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\engine\gradient_handler\_sequence_parallel_gradient_handler.py
2 4 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\getattr_handler.py
2 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\normal_pooling_handler.py
2 3 -
A
colossalai\legacy\nn\layer\colossalai_layer\normalization.py
2 2 -
A
colossalai\legacy\nn\layer\parallel_sequence\_utils.py
2 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\tensor_constructor_handler.py
2 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\unary_elementwise_handler.py
2 7 -
A
tests\kit\model_zoo\transformers\llama.py
2 8 -
A
tests\kit\model_zoo\transformers\qwen2.py
2 2 -
A
tests\kit\model_zoo\transformers\command.py
2 2 -
A
colossalai\legacy\tensor\op_wrapper.py
2 1 -
A
tests\test_shardformer\test_layer\test_dist_crossentropy.py
2 5 -
A
colossalai\booster\mixed_precision\__init__.py
2 3 -
A
colossalai\legacy\zero\shard_utils\commons.py
2 1 -
A
colossalai\cli\check\__init__.py
2 2 -
A
examples\tutorial\sequence_parallel\loss_func\bert_loss.py
2 3 -
A
examples\tutorial\sequence_parallel\model\layers\pooler.py
2 2 -
A
examples\tutorial\sequence_parallel\model\layers\dropout.py
2 2 -