main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\_analyzer\fx\symbolic_profile.py
3 2 -
A
colossalai\inference\server\utils.py
3 1 -
A
colossalai\kernel\jit\bias_dropout_add.py
3 3 -
A
colossalai\kernel\triton\fused_rotary_embedding.py
3 1 -
A
tests\test_infer\test_config_and_struct.py
3 1 -
A
tests\test_fp8\test_fp8_all_to_all_single.py
3 2 -
A
tests\kit\model_zoo\transformers\mixtral.py
3 4 -
A
tests\kit\model_zoo\transformers\mistral.py
3 3 -
A
colossalai\kernel\triton\qkv_matmul_kernel.py
3 2 -
A
tests\kit\model_zoo\diffusers\diffusers.py
3 3 -
A
colossalai\legacy\zero\__init__.py
3 2 -
A
colossalai\legacy\nn\layer\parallel_2p5d\_utils.py
3 3 -
A
tests\test_device\test_alpha_beta.py
3 3 -
A
tests\test_checkpoint_io\utils.py
3 2 -
A
colossalai\legacy\nn\metric\accuracy_3d.py
3 3 -
A
colossalai\legacy\amp\torch_amp\__init__.py
3 2 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\dummy_dataset.py
3 1 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_evoformer_stack.py
3 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\batch_norm_handler.py
3 5 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_extramsa_block.py
3 4 -