main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\sam.py
9 4 -
A
colossalai\fx\profiler\shard_utils.py
11 4 -
A
colossalai\fx\tracer\_meta_trace.py
9 5 -
A
colossalai\tensor\moe_tensor\api.py
11 5 -
A
colossalai\testing\pytest_wrapper.py
3 5 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_module\conv.py
9 5 -
A
colossalai\utils\timer.py
26 5 -
A
colossalai\fx\tracer\experimental.py
146 5 -
A
colossalai\zero\gemini\gemini_optimizer.py
153 5 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
colossalai\zero\gemini\memory_tracer\runtime_mem_tracer.py
19 5 -
A
applications\ColossalChat\coati\trainer\orpo.py
20 5 -
A
colossalai\fx\tracer\meta_patch\patched_module\convolution.py
6 5 -
A
colossalai\zero\low_level\bookkeeping\tensor_bucket.py
23 5 -
A
colossalai\fx\tracer\meta_patch\patched_module\normalization.py
4 5 -
A
colossalai\zero\wrapper.py
6 5 -
A
tests\test_fx\test_pipeline\test_hf_model\test_gpt.py
2 5 -
A
applications\ColossalChat\coati\trainer\sft.py
33 5 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_transformer_utils.py
31 5 -
A
colossalai\__init__.py
 5 -