main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\placeholder_generator.py
9 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_tensor_constructor.py
5 5 -
A
colossalai\legacy\nn\layer\parallel_2d\_operation.py
81 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_view_handler.py
22 3 -
A
tests\test_checkpoint_io\test_safetensors_async_io.py
13 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\utils.py
26 14 -
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 5 -
A
tests\test_autochunk\test_autochunk_alphafold\benchmark_autochunk_alphafold.py
29 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\softmax_generator.py
14 2 -
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_3d_plugin.py
29 3 -
A
tests\test_booster\test_mixed_precision\test_fp16_torch.py
7 3 -
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\trainer\base.py
39 3 -
A
applications\ColossalQA\colossalqa\text_splitter\chinese_text_splitter.py
 2 -
A
colossalai\fx\profiler\shard_utils.py
11 4 -
A
colossalai\tensor\d_tensor\api.py
33 6 -