main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_fx\test_pipeline\test_topo\topo_utils.py
18 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\parallel_cached_embedding.py
17 3 -
A
colossalai\nn\optimizer\lamb.py
17 4 -
A
extensions\csrc\kernel\cuda\attention\attention_utils.h
17 1 -
A
tests\test_shardformer\test_model\test_shard_mixtral.py
17 14 -
A
tests\test_pipeline\test_p2p_communication.py
17 8 -
A
examples\images\diffusion\scripts\train_searcher.py
17 2 -
A
colossalai\legacy\inference\serving\ray_serve\Colossal_Inference_rayserve.py
17 2 -
A
colossalai\auto_parallel\offload\region.py
17 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_addbmm_handler.py
17 3 -
A
colossalai\legacy\inference\async_manager.py
17 2 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_binary_elementwise_handler.py
17 3 -
A
tests\test_checkpoint_io\test_torch_fsdp_checkpoint_io.py
17 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\sum_generator.py
17 2 -
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 -
A
examples\language\gpt\hybridparallelism\benchmark.py
17 3 -
A
colossalai\cli\launcher\hostinfo.py
17 5 -
A
colossalai\cluster\device_mesh_manager.py
17 3 -
A
examples\tutorial\hybrid_parallel\train.py
17 5 -