main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\booster\mixed_precision\fp16_torch.py
14 2 -
A
examples\images\diffusion\ldm\modules\midas\api.py
14 2 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_diffuser.py
13 8 -
A
examples\language\data_utils.py
13 2 1
A
examples\tutorial\sequence_parallel\data\datasets\ict_dataset.py
13 2 -
A
tests\test_legacy\test_layers\test_3d\checks_3d\check_layer_3d.py
13 4 -
A
colossalai\inference\modeling\models\nopadding_baichuan.py
13 5 -
A
applications\ColossalChat\benchmarks\ray\mmmt_dummy.py
13 1 3
A
tests\test_legacy\test_trainer\test_pipeline\test_p2p.py
13 6 -
A
applications\ColossalChat\examples\community\peft\easy_models.py
13 1 -
A
colossalai\legacy\amp\naive_amp\_utils.py
13 2 -
A
colossalai\legacy\engine\gradient_handler\_pipeline_parallel_gradient_handler.py
13 4 -
A
tests\test_shardformer\test_layer\test_sequence_parallel.py
13 2 -
A
colossalai\legacy\pipeline\layer_spec.py
13 2 -
A
tests\test_legacy\test_layers\test_2p5d\checks_2p5d\check_layer_2p5d.py
13 4 -
A
colossalai\tensor\d_tensor\layout.py
13 9 -
A
extensions\pybind\layernorm\layer_norm.cpp
13 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\embedding_handler.py
13 3 -
A
colossalai\legacy\nn\layer\parallel_sequence\layers.py
13 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_compatibility_with_ddp.py
13 3 -