main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\sum_generator.py
17 2 1
A
colossalai\auto_parallel\tensor_shard\utils\sharding.py
17 4 -
A
tests\test_pipeline\test_p2p_communication.py
17 8 -
A
examples\images\diffusion\scripts\train_searcher.py
17 2 -
A
colossalai\cli\launcher\hostinfo.py
17 5 -
A
colossalai\cluster\device_mesh_manager.py
17 3 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
17 2 -
A
examples\language\gpt\hybridparallelism\benchmark.py
17 3 -
A
colossalai\nn\optimizer\hybrid_adam.py
17 8 1
A
colossalai\fx\profiler\memory_utils.py
16 3 -
A
examples\language\bert\benchmark.py
16 3 -
A
examples\language\deepseek\benchmark.py
16 3 3
A
tests\test_auto_parallel\test_tensor_shard\test_find_repeat_block.py
16 4 -
A
applications\ColossalChat\coati\trainer\kto.py
16 4 2
A
examples\language\gpt\experiments\pipeline_parallel\model_zoo.py
16 2 1
A
colossalai\inference\modeling\layers\diffusion.py
16 1 -
A
tests\test_legacy\test_moe\test_moe_load_balance.py
16 1 -
A
tests\test_legacy\test_comm\test_object_list_p2p_v2.py
16 4 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit_utils.py
16 5 -
A
tests\test_legacy\test_pipeline\test_middleware_1f1b.py
16 2 -