main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 5 -
A
colossalai\legacy\utils\checkpoint\module_checkpoint.py
31 2 -
A
colossalai\inference\utils.py
31 6 -
A
applications\ColossalChat\examples\community\peft\train_peft_prompts.py
31 1 -
A
colossalai\utils\safetensors.py
31 6 -
A
examples\images\diffusion\scripts\knn2img.py
31 4 -
A
colossalai\legacy\moe\openmoe\train.py
31 1 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_transformer_utils.py
31 5 -
A
tests\test_autochunk\test_autochunk_transformer\benchmark_autochunk_transformer.py
31 4 -
A
tests\test_autochunk\test_autochunk_diffuser\benchmark_autochunk_diffuser.py
31 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_permute_and_transpose_handler.py
31 3 -
A
colossalai\legacy\zero\gemini\ophooks\runtime_mem_tracer_hook.py
31 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\batch_norm_generator.py
31 12 -
A
applications\ColossalChat\coati\trainer\callbacks\performance_evaluator.py
31 1 -
A
examples\images\diffusion\ldm\modules\midas\midas\transforms.py
30 2 -
A
examples\community\roberta\pretraining\nvidia_bert_dataset_provider.py
30 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding_tablewise.py
30 3 -
A
colossalai\legacy\utils\profiler\legacy\prof_utils.py
30 2 -
A
examples\community\fp8\mnist\main.py
30 2 -
A
colossalai\shardformer\policies\vit.py
30 9 -