main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\interface\optimizer.py
19 3 -
A
tests\test_shardformer\test_flash_attention.py
19 3 -
A
colossalai\zero\gemini\memory_tracer\runtime_mem_tracer.py
19 5 -
A
applications\ColossalMoE\train.py
19 5 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
A
examples\language\bert\data.py
19 4 -
A
examples\images\diffusion\ldm\lr_scheduler.py
19 2 -
A
colossalai\zero\low_level\bookkeeping\bucket_store.py
18 3 -
A
tests\test_fx\test_codegen\test_nested_activation_checkpoint_codegen.py
18 4 -
A
tests\test_fx\test_codegen\test_activation_checkpoint_codegen.py
18 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_matmul_handler.py
18 5 -
A
applications\ColossalChat\coati\trainer\rm.py
18 4 -
A
colossalai\legacy\moe\manager.py
18 1 -
A
colossalai\legacy\inference\dynamic_batching\req_queue.py
18 1 -
A
applications\ColossalChat\examples\inference\web_chatbot\utils.py
18 1 -
A
applications\ColossalChat\coati\distributed\comm.py
18 1 -
A
colossalai\nn\optimizer\cpu_adam.py
18 2 -
A
colossalai\legacy\context\process_group_initializer\initializer_2d.py
18 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\linear_handler.py
18 12 -
A
tests\test_legacy\test_tensor\common_utils\_utils.py
18 2 -