main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\utils\profiler\profiler.py
23 1 -
A
colossalai\fx\profiler\opcount.py
23 3 -
A
colossalai\utils\model\lazy_init_context.py
23 4 -
A
colossalai\fx\passes\experimental\adding_shape_consistency_pass.py
23 1 -
A
tests\test_utils\test_zero_gradient_clippling.py
22 1 -
A
setup.py
22 3 -
A
colossalai\nn\graph\graph_node.py
22 1 -
A
colossalai\gemini\chunk\search_utils.py
22 3 -
A
colossalai\tensor\colo_parameter.py
22 2 -
A
colossalai\nn\layer\moe\routers.py
22 1 -
A
tests\test_fx\test_codegen\test_offload_codegen.py
22 3 1
A
colossalai\nn\layer\moe\layers.py
22 3 -
A
tests\test_zero\test_sharded_optim_v2.py
21 1 -
A
colossalai\auto_parallel\solver\strategy\getitem_generator.py
21 1 -
A
colossalai\fx\graph_module.py
21 2 1
A
colossalai\nn\optimizer\fused_adam.py
20 2 -
A
tests\test_fx\test_ckpt_solvers\test_ckpt_torchvision.py
20 9 1
A
colossalai\zero\sharded_param\sharded_param.py
20 1 -
A
colossalai\pipeline\rpc\utils.py
20 2 -
A
.github\workflows\scripts\build_colossalai_wheel.py
20 3 -