main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\cuda_native\csrc\scaled_masked_softmax.h
93 2 4
A
colossalai\kernel\cuda_native\csrc\scaled_upper_triang_masked_softmax.h
94 2 5
A
colossalai\gemini\chunk\chunk.py
97 3 -
A
colossalai\zero\sharded_model\sharded_model_v2.py
113 3 1
A
colossalai\fx\passes\algorithms\dynamic_programs.c
115 1 3
A
colossalai\nn\parallel\data_parallel.py
116 12 1
A
colossalai\fx\passes\algorithms\ckpt_solver_pofo.py
118 3 2
A
colossalai\utils\common.py
123 3 4
B
colossalai\fx\codegen\activation_checkpoint_codegen.py
131 14 4
A
colossalai\engine\schedule\_pipeline_schedule.py
155 2 3
A
colossalai\pipeline\rpc\_pipeline_base.py
174 4 2