main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\utils\profiler\profiler.py
23 2 -
A
colossalai\auto_parallel\tensor_shard\utils\broadcast.py
23 7 -
A
colossalai\auto_parallel\tensor_shard\utils\reshape.py
23 4 -
A
colossalai\zero\low_level\bookkeeping\tensor_bucket.py
23 5 -
A
colossalai\fx\profiler\profiler.py
23 17 -
A
tests\test_shardformer\test_model\test_shard_whisper.py
23 5 -
A
tests\test_shardformer\test_model\test_shard_t5.py
23 2 -
A
examples\images\diffusion\scripts\img2img.py
23 7 -
A
colossalai\shardformer\examples\data.py
23 2 -
A
colossalai\legacy\engine\schedule\_pipeline_schedule_v2.py
23 6 -
A
colossalai\fx\passes\experimental\adding_shape_consistency_pass.py
23 3 -
A
colossalai\legacy\zero\gemini\tensor_utils.py
23 2 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_alphafold_utils.py
22 4 -
A
colossalai\tensor\colo_parameter.py
22 5 -
A
colossalai\inference\struct.py
22 2 -
A
colossalai\legacy\nn\layer\parallel_3d\_utils.py
22 3 -
A
colossalai\inference\server\api_server.py
22 2 -
A
colossalai\auto_parallel\offload\amp_optimizer.py
22 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_view_handler.py
22 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_gpt\gpt_modules.py
22 3 -