main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\nn\lr_scheduler\multistep.py
5 5 -
A
examples\language\palm\palm_pytorch\palm_pytorch.py
21 5 -
A
applications\ColossalChat\coati\trainer\orpo.py
22 5 -
A
examples\tutorial\hybrid_parallel\config.py
 5 -
A
examples\language\palm\train.py
16 5 -
A
colossalai\nn\optimizer\fused_adam.py
20 5 -
A
colossalai\nn\lr_scheduler\linear.py
5 5 -
A
colossalai\logging\logger.py
21 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_tensor_constructor.py
5 5 -
A
tests\kit\model_zoo\torchrec\torchrec.py
13 5 -
A
examples\tutorial\hybrid_parallel\train.py
17 5 -
A
colossalai\fx\tracer\meta_patch\patched_module\normalization.py
4 5 -
A
colossalai\booster\plugin\torch_fsdp_plugin.py
46 5 -
A
examples\language\gpt\titans\model\pipeline_gpt1d.py
46 4 -
A
examples\language\gpt\titans\model\gpt1d.py
26 4 -
A
colossalai\_analyzer\_subclasses\_monkey_patch.py
 4 -
A
examples\language\grok-1\inference.py
 4 -
A
colossalai\_analyzer\_subclasses\flop_tensor.py
46 4 -
A
colossalai\legacy\zero\gemini\stateful_tensor_mgr.py
18 4 -
A
colossalai\_analyzer\fx\codegen.py
47 4 -