main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_fx\test_codegen\test_offload_codegen.py
22 4 -
A
tests\test_fx\test_codegen\test_nested_activation_checkpoint_codegen.py
18 4 -
A
colossalai\legacy\nn\layer\parallel_2d\_operation.py
81 4 4
A
tests\test_fx\test_codegen\test_activation_checkpoint_codegen.py
18 4 -
A
colossalai\auto_parallel\offload\solver.py
72 4 -
A
tests\test_fx\test_meta\test_meta_trace.py
4 4 -
A
applications\ColossalChat\examples\training_scripts\train_kto.py
46 4 3
A
colossalai\legacy\nn\layer\parallel_2p5d\_operation.py
84 4 4
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\placeholder_generator.py
9 4 -
A
tests\kit\model_zoo\transformers\vit.py
3 4 -
A
applications\ColossalChat\tests\test_train.sh
 4 -
A
colossalai\legacy\nn\layer\parallel_sequence\_operation.py
8 4 -
A
colossalai\legacy\nn\layer\parallel_sequence\layers.py
13 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\unary_elementwise_generator.py
13 4 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\where_generator.py
9 4 -
A
tests\test_fx\test_parallel_1d.py
4 4 -
A
.github\workflows\cuda_ext_check_before_merge.yml
 4 -
A
colossalai\legacy\nn\loss\loss_2d.py
8 4 -
A
colossalai\legacy\nn\loss\loss_2p5d.py
8 4 -
A
colossalai\legacy\nn\loss\loss_3d.py
8 4 -