main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\solver\_utils.py
20 7 -
A
colossalai\gemini\ophooks\utils.py
20 1 -
A
colossalai\cli\launcher\hostinfo.py
20 1 1
A
colossalai\nn\optimizer\fused_sgd.py
20 2 -
A
colossalai\zero\sharded_param\sharded_param.py
20 1 -
A
colossalai\context\config.py
20 1 -
A
colossalai\auto_parallel\solver\op_handler\dot_handler_v2.py
20 4 -
A
tests\test_fx\test_ckpt_solvers\test_ckpt_torchvision.py
20 9 1
A
colossalai\nn\layer\moe\experts.py
20 2 -
A
colossalai\kernel\cuda_native\csrc\kernels\include\dropout.h
19 1 -
A
colossalai\fx\passes\algorithms\linearize.py
19 5 1
A
colossalai\fx\tracer\meta_patch\patched_function\arithmetic.py
19 3 -
A
tests\test_moe\test_moe_zero_optim.py
19 3 -
A
tests\test_fx\test_tracer\test_torchrec_model\test_dlrm_model.py
19 2 1
A
colossalai\gemini\memory_tracer\memory_monitor.py
19 1 -
A
colossalai\nn\optimizer\lars.py
19 1 -
A
colossalai\gemini\stateful_tensor_mgr.py
18 1 -
A
colossalai\auto_parallel\solver\strategy\strategy_generator.py
18 5 -
A
tests\test_tensor\common_utils\_utils.py
18 2 -
A
tests\test_fx\test_codegen\test_nested_activation_checkpoint_codegen.py
18 1 1