main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\pybind\softmax\scaled_masked_softmax.cpp
3 1 -
A
examples\language\gpt\experiments\auto_offload\run.sh
 1 -
A
colossalai\kernel\triton\llama_act_combine_kernel.py
9 2 -
A
examples\language\gpt\experiments\auto_parallel\auto_parallel_with_gpt.py
11 4 -
A
extensions\pybind\optimizer\__init__.py
 1 -
A
colossalai\kernel\triton\qkv_matmul_kernel.py
3 2 -
A
colossalai\accelerator\npu_accelerator.py
43 2 -
A
examples\language\gpt\experiments\pipeline_parallel\train_gpt_pp.py
19 7 -
A
extensions\pybind\optimizer\fused_optimizer_cuda.py
5 1 -
A
tests\kit\model_zoo\transformers\gptj.py
4 3 -
A
tests\kit\model_zoo\transformers\gpt.py
0 3 -
A
extensions\csrc\kernel\cuda\multi_tensor_scale_kernel.cu
 1 -
A
extensions\csrc\common\target.h
24 1 -
A
examples\tutorial\sequence_parallel\model\layers\head.py
9 4 -
A
examples\tutorial\sequence_parallel\data\tokenizer\__init__.py
3 2 -
A
examples\tutorial\opt\inference\benchmark\locustfile.py
4 2 -
A
examples\language\grok-1\run_inference_fast.sh
 3 -
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 -