main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_alphafold_utils.py
22 4 -
A
examples\tutorial\sequence_parallel\data\bert_helper.py
22 3 -
A
examples\language\gpt\experiments\auto_parallel\gpt_modules.py
22 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_gpt\gpt_modules.py
22 3 -
A
colossalai\fx\graph_module.py
22 5 -
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_shardformer\test_model\test_shard_command.py
22 3 -
A
tests\test_shardformer\test_hybrid_parallel_grad_clip_norm\test_zero_optimizer.py
22 5 -
A
tests\test_fx\test_codegen\test_offload_codegen.py
22 4 -
A
tests\test_booster\test_plugin\test_low_level_zero_plugin.py
22 3 -
A
colossalai\logging\logger.py
21 5 -
A
examples\images\vit\vit_benchmark.py
21 6 -
A
colossalai\auto_parallel\meta_profiler\shard_metainfo.py
21 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_split_handler.py
21 3 -
A
colossalai\inference\modeling\models\stablediffusion3.py
21 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\embedding_generator.py
21 2 -
A
examples\community\roberta\pretraining\pretrain_utils.py
21 3 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\loader.py
21 2 -
A
colossalai\zero\gemini\memory_tracer\static_memstats_collector.py
21 2 -