main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\triton\llama_act_combine_kernel.py
9 2 -
A
colossalai\amp\naive_amp\grad_scaler\constant_grad_scaler.py
2 2 -
A
colossalai\kernel\triton\qkv_matmul_kernel.py
3 2 -
A
colossalai\kernel\triton\softmax.py
10 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\__init__.py
 2 -
A
colossalai\lazy\__init__.py
 2 -
A
applications\ColossalChat\coati\dataset\conversation.py
25 2 -
A
colossalai\amp\naive_amp\mixed_precision_mixin\base.py
5 2 -
A
colossalai\lazy\pretrained.py
49 2 -
A
colossalai\legacy\amp\__init__.py
5 2 -
A
colossalai\legacy\amp\amp_type.py
 2 -
A
applications\ColossalChat\examples\data_preparation_scripts\prepare_prompt_dataset.sh
 2 -
A
colossalai\legacy\amp\apex_amp\__init__.py
1 2 -
A
colossalai\legacy\amp\apex_amp\apex_amp.py
4 2 -
A
colossalai\auto_parallel\checkpoint\__init__.py
 2 -
A
colossalai\legacy\amp\naive_amp\__init__.py
7 2 -
A
colossalai\legacy\amp\naive_amp\_utils.py
13 2 -
A
colossalai\auto_parallel\checkpoint\build_c_ext.py
 2 -
A
colossalai\legacy\amp\naive_amp\naive_amp.py
38 2 -
A
colossalai\legacy\amp\torch_amp\__init__.py
3 2 -