main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\nn\parallel\data_parallel.py
40 3 -
A
examples\tutorial\sequence_parallel\loss_func\bert_loss.py
2 3 -
A
applications\ColossalEval\colossal_eval\models\chatglm.py
29 3 -
A
.github\workflows\release_test_pypi_before_merge.yml
 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding.py
17 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding_tablewise.py
30 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding_tablewise_split_cache.py
25 3 -
A
colossalai\legacy\nn\parallel\layers\colo_module.py
12 3 -
A
colossalai\legacy\nn\parallel\layers\embedding.py
5 3 -
A
colossalai\legacy\nn\parallel\layers\linear.py
5 3 -
A
colossalai\legacy\nn\parallel\layers\module_utils.py
36 3 -
A
colossalai\booster\__init__.py
 3 -
A
examples\images\vit\run_benchmark.sh
 3 -
A
colossalai\booster\accelerator.py
5 3 -
A
colossalai\shardformer\shard\grad_ckpt_config.py
11 3 -
A
colossalai\booster\mixed_precision\__init__.py
2 3 -
A
colossalai\booster\mixed_precision\fp16_apex.py
 3 -
A
colossalai\legacy\pipeline\rpc\utils.py
33 3 -
A
colossalai\booster\mixed_precision\fp16_naive.py
1 3 -
A
colossalai\booster\plugin\__init__.py
 3 -