main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_zero\test_gemini\test_optim.py
22 12 -
A
tests\test_shardformer\test_model\test_shard_qwen2.py
22 5 -
A
tests\test_fx\test_codegen\test_offload_codegen.py
22 4 -
A
examples\inference\stable_diffusion\benchmark_sd3.py
22 1 -
A
applications\ColossalChat\coati\trainer\orpo.py
22 5 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_alphafold_utils.py
22 4 -
A
colossalai\logging\logger.py
21 5 -
A
colossalai\zero\gemini\gemini_hook.py
21 5 -
A
colossalai\zero\gemini\memory_tracer\memory_stats.py
21 4 -
A
colossalai\auto_parallel\meta_profiler\shard_metainfo.py
21 2 -
A
examples\tutorial\new_api\cifar_vit\train.py
21 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\embedding_generator.py
21 2 -
A
colossalai\inference\modeling\models\stablediffusion3.py
21 2 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\loader.py
21 2 -
A
examples\images\vit\vit_benchmark.py
21 6 -
A
examples\community\roberta\pretraining\pretrain_utils.py
21 3 -
A
colossalai\shardformer\policies\base_policy.py
21 11 -
A
colossalai\legacy\inference\dynamic_batching\sampling_params.py
21 1 -
A
examples\language\palm\palm_pytorch\palm_pytorch.py
21 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_split_handler.py
21 3 -