main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\booster\mixed_precision\fp16_naive.py
1 3 -
A
colossalai\legacy\utils\memory.py
24 3 1
A
colossalai\legacy\utils\profiler\legacy\comm_profiler.py
43 3 -
A
colossalai\booster\plugin\__init__.py
 3 -
A
colossalai\booster\plugin\dp_plugin_base.py
3 3 -
A
tests\kit\model_zoo\transformers\gptj.py
4 3 -
A
examples\images\diffusion\scripts\download_first_stages.sh
 3 -
A
.github\workflows\doc_build_on_schedule_after_release.yml
 3 -
A
colossalai\legacy\zero\gemini\__init__.py
 3 -
A
colossalai\booster\plugin\plugin_base.py
12 3 -
A
colossalai\booster\plugin\pp_plugin_base.py
1 3 -
A
examples\tutorial\auto_parallel\auto_ckpt_solver_test.py
9 3 -
A
colossalai\legacy\zero\gemini\tensor_placement_policy.py
24 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_embedding_handler.py
14 3 -
A
tests\test_analyzer\test_subclasses\test_aten.py
6 3 -
A
tests\kit\model_zoo\transformers\mistral.py
3 3 -
A
examples\images\vit\data.py
9 3 -
A
applications\ColossalEval\colossal_eval\models\chatglm.py
29 3 1
A
colossalai\legacy\zero\shard_utils\bucket_tensor_shard_strategy.py
11 3 -
A
colossalai\legacy\zero\shard_utils\tensor_shard_strategy.py
9 3 -