main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\normal_pooling_handler.py
2 3 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_opt.py
2 13 -
A
tests\test_fx\test_pipeline\test_hf_model\test_gpt.py
2 5 -
A
examples\tutorial\sequence_parallel\loss_func\bert_loss.py
2 3 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\tensor_constructor_handler.py
2 2 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\unary_elementwise_handler.py
2 7 -
A
colossalai\legacy\engine\gradient_handler\_base_gradient_handler.py
2 2 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit.py
2 3 -
A
examples\images\diffusion\ldm\modules\midas\midas\base_model.py
2 2 -
A
examples\images\diffusion\scripts\inpaint.py
2 4 -
A
colossalai\booster\mixed_precision\__init__.py
2 3 -
A
examples\inference\llama\llama_generation.py
2 3 -
A
colossalai\cli\check\__init__.py
2 2 -
A
colossalai\legacy\nn\layer\parallel_sequence\_utils.py
2 1 -
A
colossalai\legacy\nn\layer\colossalai_layer\normalization.py
2 2 -
A
colossalai\fx\_compatibility.py
2 4 -
A
examples\images\diffusion\scripts\tests\test_watermark.py
2 2 -
A
applications\ColossalChat\coati\quant\utils.py
2 1 -