main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\cuda_native\mha\flash_attn_2.py
6 1 -
A
applications\Colossal-LLaMA-2\colossal_llama2\__init__.py
 1 -
A
colossalai\kernel\cuda_native\mha\mha.py
18 1 -
A
colossalai\auto_parallel\offload\training_simulator.py
72 1 -
A
colossalai\kernel\cuda_native\multihead_attention.py
30 1 -
A
applications\Chat\coati\models\gpt\gpt_actor.py
4 1 -
A
colossalai\kernel\jit\__init__.py
 1 -
A
colossalai\auto_parallel\passes\comm_metainfo_pass.py
12 1 -
A
colossalai\kernel\jit\bias_gelu.py
4 1 -
A
applications\Colossal-LLaMA-2\colossal_llama2\dataset\conversation.py
14 1 -
A
colossalai\fx\tracer\meta_patch\patched_module\pooling.py
12 1 -
A
colossalai\_analyzer\fx\__init__.py
 1 -
A
colossalai\fx\tracer\meta_patch\patched_module\rnn.py
2 1 -
A
colossalai\fx\tracer\registry.py
4 1 -
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 1 1
A
.github\ISSUE_TEMPLATE\bug-report.yml
 1 -
A
.github\ISSUE_TEMPLATE\documentation.yml
 1 -
A
.github\workflows\build_on_schedule.yml
 1 -
A
.github\workflows\doc_test_on_pr.yml
 1 -
A
.github\workflows\scripts\check_doc_i18n.py
11 1 -