main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_tensor_constructor.py
5 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_where_handler.py
5 5 -
A
colossalai\legacy\nn\layer\parallel_2d\layers.py
 5 -
A
tests\test_autochunk\test_autochunk_alphafold\benchmark_autochunk_alphafold.py
29 5 -
A
applications\ColossalEval\colossal_eval\dataset\__init__.py
 5 -
A
tests\test_cluster\test_process_group_mesh.py
7 5 -
A
colossalai\legacy\nn\layer\parallel_2p5d\layers.py
 5 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_gpt.py
2 5 -
A
tests\test_autochunk\test_autochunk_transformer\test_autochunk_transformer_utils.py
31 5 -
A
tests\test_autochunk\test_autochunk_vit\test_autochunk_vit_utils.py
16 5 -
A
tests\test_booster\test_accelerator.py
1 5 -
A
colossalai\legacy\nn\layer\parallel_3d\layers.py
 5 -
A
tests\test_checkpoint_io\test_gemini_checkpoint_io.py
25 5 -
A
tests\test_optimizer\test_nvme.py
7 5 -
A
tests\test_optimizer\test_adam_kernel.py
19 5 -
A
tests\test_moe\test_mixtral_layer.py
6 5 -
A
tests\test_legacy\test_utils\test_memory.py
3 5 -
A
tests\test_legacy\test_comm\test_comm.py
8 5 -
A
colossalai\legacy\nn\layer\vanilla\layers.py
38 5 -
A
colossalai\autochunk\select_chunk.py
44 5 -