main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalChat\examples\inference\web_chatbot\utils.py
18 1 -
A
examples\tutorial\sequence_parallel\lr_scheduler\annealing_lr.py
20 2 -
A
colossalai\legacy\engine\gradient_handler\utils.py
8 1 -
A
examples\tutorial\sequence_parallel\model\layers\__init__.py
 2 -
A
colossalai\auto_parallel\meta_profiler\meta_registry\where.py
4 3 -
A
examples\tutorial\sequence_parallel\model\layers\dropout.py
2 2 -
A
colossalai\legacy\engine\schedule\_base_schedule.py
37 3 -
A
examples\tutorial\sequence_parallel\model\layers\head.py
9 4 -
A
tests\kit\model_zoo\registry.py
12 8 -
A
tests\test_fp8\test_fp8_all_to_all_single.py
3 2 -
A
tests\test_fp8\test_fp8_all_to_all.py
6 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_norm_metainfo.py
3 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_bias_addition_forward.py
7 3 -
A
tests\test_analyzer\test_fx\test_symbolic_profile.py
6 4 -
A
tests\kit\model_zoo\transformers\mistral.py
3 3 -
A
tests\kit\model_zoo\custom\simple_mlp.py
7 2 -
A
extensions\csrc\kernel\cuda\convert_fp8_kernel.cu
 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -