main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\initialize.py
11 9 -
A
colossalai\interface\__init__.py
 3 -
A
colossalai\interface\model.py
46 3 -
A
colossalai\interface\optimizer.py
19 3 -
A
colossalai\interface\pretrained.py
2 1 -
A
colossalai\kernel\jit\__init__.py
 2 -
A
colossalai\kernel\jit\bias_dropout_add.py
3 3 -
A
colossalai\kernel\jit\bias_gelu.py
4 6 -
A
colossalai\kernel\jit\option.py
10 5 -
A
colossalai\kernel\kernel_loader.py
7 5 -
A
colossalai\kernel\triton\__init__.py
 11 -
B
colossalai\kernel\triton\context_attn_unpad.py
26 1 4
A
colossalai\kernel\triton\flash_decoding.py
18 3 -
A
colossalai\kernel\triton\fused_rotary_embedding.py
3 1 -
A
colossalai\kernel\triton\kvcache_copy.py
14 1 -
A
colossalai\kernel\triton\llama_act_combine_kernel.py
9 2 -
A
colossalai\kernel\triton\no_pad_rotary_embedding.py
15 1 1
A
colossalai\kernel\triton\qkv_matmul_kernel.py
3 2 -
A
colossalai\kernel\triton\rms_layernorm.py
10 1 -
A
colossalai\kernel\triton\rotary_cache_copy.py
6 1 -