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
applications\ColossalChat\coati\distributed\inference_backend.py
19 2 2
A
extensions\csrc\kernel\cuda\multi_tensor_lamb_kernel.cu
 1 1
A
extensions\csrc\kernel\cuda\multi_tensor_l2norm_kernel.cu
 1 1
A
extensions\csrc\kernel\cuda\attention\attention_utils.h
17 1 -
A
extensions\csrc\kernel\arm\cpu_adam_arm.h
33 2 1
A
extensions\csrc\funcs\reduce_function.h
4 1 -
A
examples\tutorial\sequence_parallel\train.py
36 5 -
A
examples\tutorial\sequence_parallel\model\layers\mlp.py
3 2 -
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 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 -