main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_tensor\model\test_gpt2.py
38 3 -
A
colossalai\context\process_group_initializer\initializer_2p5d.py
37 1 -
A
colossalai\amp\naive_amp\naive_amp.py
37 1 1
A
colossalai\engine\schedule\_base_schedule.py
37 1 -
A
colossalai\engine\gradient_accumulation\_gradient_accumulation.py
37 1 -
A
colossalai\kernel\cuda_native\scaled_softmax.py
36 1 -
A
colossalai\nn\parallel\layers\module_utils.py
36 2 1
A
colossalai\pipeline\pipeline_process_group.py
35 2 -
A
tests\test_tensor\model\test_module_spec.py
34 1 1
A
colossalai\utils\rank_recorder\rank_recorder.py
34 1 -
A
colossalai\zero\init_ctx\init_context.py
33 2 -
A
colossalai\fx\passes\split_module.py
33 2 1
A
colossalai\nn\layer\vanilla\layers.py
33 2 -
A
colossalai\gemini\stateful_tensor.py
33 1 -
A
colossalai\fx\passes\adding_split_node_pass.py
33 3 -
A
colossalai\utils\profiler\stateful_tensor_mem_extention.py
32 2 -
A
colossalai\device\device_mesh.py
31 3 -
A
colossalai\utils\checkpoint\module_checkpoint.py
31 10 1
A
colossalai\communication\collective.py
31 1 -
A
colossalai\kernel\cuda_native\multihead_attention.py
31 1 -