main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalChat\coati\experience_maker\base.py
8 1 -
A
extensions\pybind\optimizer\__init__.py
 1 -
A
colossalai\legacy\inference\tensor_parallel\kvcache_manager.py
12 2 -
A
extensions\pybind\optimizer\optimizer.cpp
1 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\registry.py
4 5 -
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
colossalai\legacy\inference\tensor_parallel\modeling\_utils.py
9 2 -
A
extensions\pybind\softmax\scaled_upper_triang_masked_softmax.cpp
3 1 -
A
applications\ColossalEval\colossal_eval\dataset\base.py
6 3 -
A
tests\kit\model_zoo\torchvision\__init__.py
 1 -
A
tests\kit\model_zoo\torchrec\torchrec.py
13 5 -
A
tests\kit\model_zoo\custom\simple_mlp.py
7 1 -
A
tests\kit\model_zoo\custom\nested_model.py
7 1 -
A
.github\workflows\doc_test_on_pr.yml
 8 -
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\__init__.py
 1 -
A
applications\ColossalQA\tests\test_text_splitter.py
 1 -
A
colossalai\fx\tracer\meta_patch\patched_function\activation_function.py
1 3 -
A
colossalai\utils\multi_tensor_apply\__init__.py
 1 -