main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\quantization\fp8.py
110 6 -
A
colossalai\tensor\d_tensor\api.py
33 6 -
A
colossalai\nn\lr_scheduler\cosine.py
6 6 -
A
colossalai\device\alpha_beta_profiler.py
36 6 -
A
colossalai\cli\launcher\multinode_runner.py
16 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\bmm_handler.py
9 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\batch_norm_handler.py
3 5 -
A
examples\language\grok-1\inference_tp.py
 5 -
A
examples\language\gpt\titans\train_gpt.py
12 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\conv_handler.py
15 5 -
A
examples\language\gpt\test_ci.sh
 5 -
A
examples\language\gpt\titans\model\embed.py
71 5 -
A
colossalai\cli\launcher\hostinfo.py
17 5 -
A
colossalai\auto_parallel\passes\meta_info_prop.py
32 5 -
A
examples\language\gpt\experiments\auto_offload\train_gpt_offload.py
6 5 -
A
colossalai\legacy\nn\layer\parallel_3d\layers.py
 5 -
A
examples\images\dreambooth\train_dreambooth_colossalai_lora.py
74 5 -
A
examples\images\dreambooth\train_dreambooth_colossalai.py
70 5 -
A
colossalai\legacy\nn\layer\parallel_2p5d\layers.py
 5 -
A
colossalai\legacy\nn\layer\vanilla\layers.py
38 5 -