main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\engine\gradient_handler\_pipeline_parallel_gradient_handler.py
13 2 -
A
colossalai\amp\naive_amp\_utils.py
13 1 -
A
tests\test_layers\test_2p5d\checks_2p5d\check_layer_2p5d.py
13 1 -
A
colossalai\fx\passes\meta_info_prop.py
13 13 -
A
.github\workflows\scripts\generate_release_draft.py
13 1 -
A
colossalai\cli\benchmark\benchmark.py
13 1 -
A
colossalai\fx\profiler\experimental\profiler_function\arithmetic.py
13 1 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\strided_batch_gemm.h
13 1 -
A
tests\test_gemini\update\test_chunkv2.py
13 4 -
A
colossalai\kernel\cuda_native\csrc\layer_norm_cuda.cpp
13 1 -
A
tests\test_zero\test_zero_engine.py
13 1 -
A
tests\test_utils\test_norm_gradient_clipping.py
13 2 -
A
tests\test_trainer\test_pipeline\test_p2p.py
13 1 -
A
tests\test_tensor\core\test_tensor.py
13 2 -
A
tests\test_pipeline\rpc_test_utils.py
13 5 -
A
colossalai\nn\layer\parallel_sequence\layers.py
13 2 -
A
tests\test_layers\test_2d\checks_2d\check_operation_2d.py
12 1 -
A
colossalai\nn\_ops\addmm.py
12 3 -
A
colossalai\nn\optimizer\colossalai_optimizer.py
12 2 -
A
colossalai\fx\passes\shard_1d_pass.py
12 4 -