main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\nn\layer\parallel_2p5d\layers.py
 2 -
A
colossalai\tensor\colo_parameter.py
20 2 -
A
colossalai\legacy\nn\layer\parallel_1d\layers.py
 2 -
A
colossalai\legacy\tensor\process_group.py
53 2 1
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 2 -
A
colossalai\legacy\initialize.py
60 2 1
A
colossalai\tensor\d_tensor\api.py
31 2 -
A
colossalai\legacy\engine\gradient_handler\__init__.py
 2 -
A
colossalai\legacy\engine\schedule\_pipeline_schedule.py
155 2 4
A
colossalai\legacy\context\parallel_context.py
69 2 -
A
colossalai\legacy\utils\activation_checkpoint.py
42 2 1
A
colossalai\legacy\engine\schedule\_pipeline_schedule_v2.py
23 2 -
A
colossalai\legacy\amp\torch_amp\torch_amp.py
9 2 -
A
colossalai\kernel\triton\smooth_attention.py
22 2 -
A
colossalai\shardformer\policies\blip2.py
11 2 -
A
colossalai\kernel\cuda_native\mha\utils.py
6 2 -
A
colossalai\kernel\cuda_native\csrc\kernels\cuda_util.cu
 2 -
A
colossalai\kernel\cuda_native\csrc\cpu_adam.cpp
66 2 3
A
colossalai\initialize.py
13 2 -
A
colossalai\inference\quant\smoothquant\models\__init__.py
 2 -