main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\csrc\kernel\arm\cpu_adam_arm.h
33 2 -
A
colossalai\utils\safetensors.py
33 9 -
A
colossalai\inference\modeling\models\pixart_alpha.py
33 2 -
A
colossalai\legacy\moe\layer\routers.py
33 2 -
A
colossalai\checkpoint_io\checkpoint_io_base.py
32 8 -
A
colossalai\legacy\zero\sharded_model\_utils.py
32 2 -
A
tests\test_optimizer\test_dist_adafactor.py
32 2 -
A
colossalai\auto_parallel\tensor_shard\sharding_strategy.py
32 13 -
A
examples\community\roberta\preprocessing\tokenize_mask.py
32 2 -
A
colossalai\legacy\utils\profiler\stateful_tensor_mem_extention.py
32 2 -
A
applications\ColossalEval\colossal_eval\evaluate\dataset_evaluator\gpt_judge.py
32 1 -
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 5 -
A
colossalai\legacy\moe\openmoe\train.py
31 1 -
A
applications\ColossalChat\coati\trainer\callbacks\performance_evaluator.py
31 1 -
A
colossalai\checkpoint_io\general_checkpoint_io.py
31 14 -
A
applications\ColossalChat\examples\community\peft\train_peft_prompts.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 -