main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\cluster\dist_coordinator.py
18 4 -
A
colossalai\context\__init__.py
 4 -
A
colossalai\nn\optimizer\fused_lamb.py
29 4 -
A
colossalai\nn\optimizer\fused_sgd.py
20 4 -
A
applications\ColossalChat\coati\models\utils.py
18 4 -
A
colossalai\device\__init__.py
 4 -
A
colossalai\nn\optimizer\lamb.py
17 4 -
A
colossalai\nn\optimizer\lars.py
19 4 -
A
applications\ColossalEval\examples\gpt_evaluation\inference.py
37 4 2
A
colossalai\nn\optimizer\nvme_optimizer.py
44 4 -
A
colossalai\pipeline\p2p.py
148 4 1
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_embedding_metainfo.py
1 4 -
A
colossalai\fx\_compatibility.py
2 4 -
A
applications\ColossalMoE\infer.py
 4 -
B
colossalai\fx\codegen\activation_checkpoint_codegen.py
131 4 3
A
colossalai\fx\passes\__init__.py
 4 -
A
colossalai\shardformer\examples\convergence_benchmark.py
15 4 -
A
colossalai\shardformer\examples\performance_benchmark.py
7 4 -
A
tests\kit\model_zoo\transformers\blip2.py
1 4 -
A
tests\kit\model_zoo\transformers\vit.py
3 4 -