main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\Colossal-LLaMA\colossal_llama\dataset\conversation.py
14 2 -
A
extensions\csrc\kernel\cuda\multi_tensor_apply.cuh
 1 -
A
colossalai\legacy\inference\hybridengine\modeling\__init__.py
 1 -
A
extensions\csrc\kernel\cuda\multi_tensor_lamb_kernel.cu
 1 1
A
colossalai\auto_parallel\offload\training_simulator.py
72 3 1
A
extensions\csrc\kernel\cuda\multi_tensor_sgd_kernel.cu
 1 -
A
colossalai\legacy\inference\hybridengine\modeling\llama.py
42 1 3
A
extensions\csrc\kernel\cuda\scaled_masked_softmax_kernel.cu
 1 1
A
applications\ColossalChat\examples\training_scripts\train_grpo.sh
 1 1
A
setup.py
8 9 -
A
requirements\requirements.txt
 5 -
A
extensions\pybind\moe\moe_cuda.py
5 1 -
A
extensions\pybind\moe\__init__.py
 1 -
A
extensions\pybind\inference\inference_ops_cuda.py
5 1 -
A
extensions\csrc\kernel\x86\cpu_adam.h
16 2 1
A
extensions\csrc\kernel\cuda\utils\micros.h
0 1 -
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 -