main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\embedding_generator.py
21 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_output_handler.py
6 8 -
A
colossalai\legacy\nn\layer\colossalai_layer\linear.py
5 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_placeholder_handler.py
7 7 -
A
tests\test_checkpoint_io\test_general_checkpoint_io.py
8 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_softmax_handler.py
8 3 -
A
colossalai\legacy\nn\layer\parallel_1d\__init__.py
 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_sum_handler.py
15 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getitem_generator.py
27 7 -
A
tests\test_checkpoint_io\test_low_level_zero_checkpoint_io.py
26 6 -
A
tests\test_checkpoint_io\test_hybrid_parallel_plugin_checkpoint_io.py
8 4 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit.py
2 3 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
.github\workflows\doc_test_on_pr.yml
 5 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -