main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
extensions\csrc\kernel\cuda\utils\gpu_launch_config.h
8 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\binary_elementwise_handler.py
8 7 -
A
examples\inference\benchmark_ops\benchmark_rmsnorm.py
8 1 -
A
colossalai\legacy\inference\tensor_parallel\policies\llama.py
8 1 1
A
examples\language\grok-1\grok1_policy.py
8 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getattr_generator.py
8 3 -
A
colossalai\legacy\nn\layer\parallel_sequence\_operation.py
8 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\tensor_constructor_generator.py
8 2 -
A
colossalai\legacy\nn\loss\loss_2d.py
8 4 -
A
colossalai\legacy\nn\loss\loss_2p5d.py
8 4 -
A
colossalai\legacy\nn\loss\loss_3d.py
8 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\transpose_handler.py
8 2 -
A
colossalai\shardformer\layer\dropout.py
8 10 -
A
colossalai\pipeline\weight_grad_store.py
8 2 -
A
applications\ColossalChat\coati\experience_maker\base.py
8 1 -
A
applications\Colossal-LLaMA\colossal_llama\utils\utils.py
8 1 -
A
colossalai\legacy\zero\gemini\gemini_context.py
8 2 -
A
tests\test_zero\test_gemini\test_chunk_mgrv2.py
8 4 -
A
tests\test_pipeline\test_stage_manager.py
8 4 -
A
tests\test_pipeline\test_pipeline_utils\test_whisper_pipeline_utils.py
8 2 -