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
colossalai\legacy\trainer\hooks\_metric_hook.py
71 5 -
A
examples\language\gpt\titans\model\embed.py
71 5 -
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 5 -
A
examples\language\gpt\titans\train_gpt.py
12 5 -
A
colossalai\legacy\nn\layer\parallel_2p5d\layers.py
 5 -
A
colossalai\fx\passes\concrete_info_prop.py
15 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\conv_handler.py
15 5 -
A
colossalai\fx\graph_module.py
22 5 -
A
examples\language\grok-1\inference_tp.py
 5 -
A
examples\language\gpt\test_ci.sh
 5 -
A
colossalai\legacy\nn\layer\parallel_3d\layers.py
 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\batch_norm_handler.py
3 5 -
A
applications\ColossalMoE\train.py
19 5 -
A
examples\language\gpt\experiments\auto_offload\train_gpt_offload.py
6 5 -
A
colossalai\legacy\nn\layer\parallel_1d\layers.py
 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\bmm_handler.py
9 5 -
A
examples\images\diffusion\train_colossalai.sh
 5 -
A
examples\images\diffusion\scripts\tests\test_checkpoint.py
 5 1
A
applications\ColossalQA\colossalqa\local\llm.py
38 5 -