main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\kernel\triton\flash_decoding.py
18 3 -
A
applications\ColossalEval\examples\gpt_evaluation\eval.py
18 1 -
A
applications\ColossalChat\coati\models\utils.py
18 4 -
A
colossalai\fx\passes\meta_info_prop.py
18 23 -
A
applications\ColossalChat\coati\ray\callbacks\base.py
18 1 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding.py
17 3 -
A
examples\images\diffusion\scripts\train_searcher.py
17 2 -
A
colossalai\nn\optimizer\hybrid_adam.py
17 8 1
A
tests\test_shardformer\test_model\test_shard_mixtral.py
17 14 -
A
colossalai\nn\optimizer\came.py
17 1 -
A
colossalai\legacy\inference\serving\ray_serve\Colossal_Inference_rayserve.py
17 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_addbmm_handler.py
17 3 -
A
colossalai\auto_parallel\offload\region.py
17 3 -
A
colossalai\legacy\inference\async_manager.py
17 2 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
17 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_binary_elementwise_handler.py
17 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\sum_generator.py
17 2 1
A
tests\test_pipeline\test_p2p_communication.py
17 8 -
A
colossalai\auto_parallel\tensor_shard\utils\sharding.py
17 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_gpt\test_runtime_with_gpt_modules.py
17 3 -