main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\language\palm\palm_pytorch\palm_pytorch.py
21 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\language\gpt\titans\train_gpt.py
12 5 -
A
colossalai\legacy\nn\layer\parallel_2p5d\layers.py
 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\batch_norm_handler.py
3 5 -
A
colossalai\legacy\nn\layer\parallel_3d\layers.py
 5 -
A
examples\language\grok-1\inference_tp.py
 5 -
A
examples\language\gpt\titans\model\embed.py
71 5 -
A
colossalai\fx\passes\concrete_info_prop.py
15 5 -
A
colossalai\fx\graph_module.py
22 5 -
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 5 -
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 5 -
A
examples\language\gpt\test_ci.sh
 5 -
A
examples\language\gpt\experiments\auto_offload\train_gpt_offload.py
6 5 -
A
examples\images\dreambooth\train_dreambooth_colossalai_lora.py
74 5 -
A
examples\images\diffusion\scripts\tests\test_checkpoint.py
 5 1
A
applications\ColossalMoE\train.py
19 5 -
A
colossalai\auto_parallel\offload\base_offload_module.py
25 5 -
A
examples\images\diffusion\train_colossalai.sh
 5 -