main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\registry\registry.py
12 1 -
A
colossalai\fx\profiler\experimental\profiler_module\convolution.py
12 1 -
A
tests\test_gemini\update\test_zeroddp_state_dict.py
12 3 -
A
colossalai\nn\_ops\addmm.py
12 3 -
A
colossalai\fx\tracer\_tracer_utils.py
12 4 -
A
colossalai\fx\tracer\meta_patch\patched_function\convolution.py
12 1 -
A
colossalai\nn\layer\utils\common.py
12 1 -
A
tests\test_ddp\test_ddp_state_dict.py
12 8 -
A
colossalai\fx\tracer\meta_patch\patched_module\pooling.py
12 2 -
A
colossalai\tensor\distspec.py
12 3 -
A
colossalai\gemini\ophooks\_shard_param_ophook.py
12 1 -
A
tests\test_ddp\test_ddp_ignore_params.py
12 5 -
A
tests\test_gemini\update\test_fwd_bwd.py
12 3 -
A
colossalai\auto_parallel\solver\op_handler\layer_norm_handler.py
12 1 -
A
tests\test_layers\test_2d\checks_2d\check_operation_2d.py
12 1 -
A
colossalai\nn\optimizer\colossalai_optimizer.py
12 2 -
A
tests\test_utils\test_checkpoint\test_checkpoint_3d.py
12 1 -
A
tests\test_utils\test_checkpoint\test_checkpoint_2p5d.py
12 1 -
A
tests\test_utils\test_checkpoint\test_checkpoint_2d.py
12 1 -
A
tests\test_utils\test_checkpoint\test_checkpoint_1d.py
12 1 -