main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalEval\examples\gpt_evaluation\eval.py
18 1 -
A
colossalai\legacy\context\process_group_initializer\initializer_2d.py
18 3 -
A
applications\ColossalChat\coati\ray\callbacks\base.py
18 1 -
A
examples\tutorial\opt\inference\cache.py
18 2 4
A
tests\test_legacy\test_tensor\common_utils\_utils.py
18 2 -
A
colossalai\cluster\dist_coordinator.py
18 4 -
A
applications\Colossal-LLaMA\dataset\prepare_sft_dataset.py
18 1 -
A
colossalai\fx\passes\meta_info_prop.py
18 23 -
A
colossalai\kernel\triton\flash_decoding.py
18 3 -
A
examples\tutorial\auto_parallel\bench_utils.py
18 7 -
A
applications\ColossalChat\coati\trainer\rm.py
18 4 -
A
colossalai\zero\low_level\bookkeeping\gradient_store.py
18 10 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding.py
17 3 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
17 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_gpt\test_runtime_with_gpt_modules.py
17 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_addbmm_handler.py
17 3 -
A
tests\test_pipeline\test_p2p_communication.py
17 8 -
A
colossalai\auto_parallel\offload\region.py
17 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_binary_elementwise_handler.py
17 3 -
A
colossalai\legacy\inference\serving\ray_serve\Colossal_Inference_rayserve.py
17 2 -