main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_fx\test_pipeline\test_hf_model\test_opt.py
2 7 -
A
colossalai\inference\core\engine.py
14 7 -
A
tests\test_autochunk\test_autochunk_diffuser\test_autochunk_unet.py
2 7 -
A
colossalai\shardformer\shard\sharder.py
39 7 -
A
colossalai\fx\tracer\_tracer_utils.py
12 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\reshape_generator.py
45 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\layer_norm_generator.py
26 7 -
A
tests\test_checkpoint_io\test_gemini_torch_compability.py
20 7 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_placeholder_handler.py
7 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getitem_generator.py
27 7 -
A
colossalai\fx\tracer\meta_patch\patched_function\torch_ops.py
47 7 -
A
applications\ColossalChat\examples\training_scripts\train_rm.py
40 7 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_activation_metainfo.py
1 7 -
A
tests\test_auto_parallel\test_tensor_shard\test_gpt\test_solver_with_gpt_module.py
8 7 -
A
colossalai\shardformer\modeling\chatglm2_6b\modeling_chatglm.py
 7 -
A
tests\test_zero\test_gemini\test_zeroddp_state_dict.py
11 7 -
A
tests\test_checkpoint_io\test_plugins_huggingface_compatibility.py
11 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\binary_elementwise_handler.py
8 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\unary_elementwise_handler.py
2 7 -
A
examples\language\gpt\gemini\train_gpt_demo.py
29 6 -