main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\layer\normalization.py
33 4 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\softmax_generator.py
14 2 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\sum_generator.py
17 2 1
A
applications\Colossal-LLaMA\dataset\prepare_sft_dataset.py
18 1 1
A
colossalai\legacy\inference\tensor_parallel\policies\llama.py
8 1 1
A
colossalai\kernel\triton\no_pad_rotary_embedding.py
15 1 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\unary_elementwise_generator.py
13 4 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\reshape_generator.py
45 7 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\layer_norm_generator.py
26 7 1
A
colossalai\_analyzer\fx\tracer\proxy.py
27 2 1
A
colossalai\auto_parallel\offload\training_simulator.py
72 3 1
A
colossalai\_analyzer\fx\codegen.py
47 4 1
A
applications\ColossalMoE\utils.py
9 1 1
A
applications\ColossalMoE\train.py
19 5 1
A
applications\Colossal-LLaMA\train.example.sh
 2 1
A
extensions\csrc\kernel\cuda\context_kv_cache_memcpy_kernel.cu
 1 1
A
extensions\csrc\kernel\arm\cpu_adam_arm.h
33 2 1
A
applications\ColossalEval\colossal_eval\models\vllm.py
63 1 1
A
applications\ColossalEval\colossal_eval\models\huggingface.py
73 9 1
A
applications\Colossal-LLaMA\dataset\prepare_pretrain_dataset.py
16 1 1