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_safetensors_async_io.py
13 5 -
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_torch_fsdp_checkpoint_io.py
17 2 -
A
tests\test_checkpoint_io\test_torch_ddp_checkpoint_io.py
5 3 -
A
tests\test_booster\test_plugin\test_gemini_plugin.py
24 2 -
A
tests\test_booster\test_plugin\test_3d_plugin.py
29 3 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit.py
2 3 -
A
tests\test_autochunk\test_autochunk_alphafold\benchmark_autochunk_alphafold.py
29 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_view_handler.py
22 3 -
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 -