main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\registry.py
4 5 -
A
examples\language\performance_evaluator.py
20 5 -
A
examples\language\grok-1\inference_tp.py
 5 -
A
examples\language\gpt\titans\model\embed.py
71 5 -
A
applications\ColossalMoE\train.py
19 5 -
A
colossalai\legacy\nn\layer\vanilla\layers.py
38 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\conv_handler.py
15 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\bmm_handler.py
9 5 -
A
examples\tutorial\sequence_parallel\train.py
36 5 -
A
colossalai\legacy\trainer\hooks\_metric_hook.py
71 5 -
A
examples\language\gpt\test_ci.sh
 5 -
A
examples\language\gpt\experiments\auto_offload\train_gpt_offload.py
6 5 -
A
colossalai\fx\graph_module.py
22 5 -
A
.github\workflows\run_chatgpt_examples.yml
 5 -
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 5 -
A
examples\images\diffusion\train_colossalai.sh
 5 -
A
applications\ColossalQA\colossalqa\local\llm.py
38 5 -
A
colossalai\legacy\nn\layer\parallel_1d\layers.py
 5 -
A
colossalai\device\device_mesh.py
56 5 -
A
examples\images\dreambooth\train_dreambooth_colossalai.py
70 5 -