main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\sum_handler.py
12 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_bias_linear_module_node.py
7 3 -
A
colossalai\legacy\nn\layer\parallel_1d\layers.py
 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_bmm_handler.py
11 3 -
A
applications\ColossalEval\colossal_eval\dataset\safetybench_en.py
 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_default_reshape_handler.py
4 4 -
A
colossalai\legacy\nn\layer\parallel_2d\_operation.py
81 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_getattr_handler.py
5 6 -
A
colossalai\auto_parallel\tensor_shard\node_handler\transpose_handler.py
8 2 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_extramsa_block.py
3 4 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_evoformer_stack.py
3 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_softmax_handler.py
8 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_shard_option.py
9 5 -
A
.github\workflows\doc_test_on_pr.yml
 5 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\utils\ckpt_io.py
7 1 -
A
applications\ColossalQA\tests\test_memory.py
7 1 -
A
colossalai\fx\tracer\meta_patch\__init__.py
 3 -
A
colossalai\utils\__init__.py
 2 -