main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\images\vit\data.py
9 3 -
A
colossalai\inference\kv_cache\block_cache.py
9 1 -
A
tests\test_fx\test_pipeline\test_hf_model\hf_utils.py
9 2 -
A
.github\workflows\scripts\update_setup_for_nightly.py
9 1 -
A
colossalai\kernel\triton\llama_act_combine_kernel.py
9 2 -
A
colossalai\amp\naive_amp\grad_scaler\base_grad_scaler.py
9 4 -
A
colossalai\legacy\amp\torch_amp\torch_amp.py
9 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_shard_option.py
9 5 -
A
tests\test_fx\test_tracer\test_timm_model\test_timm_model.py
9 17 -
A
colossalai\legacy\inference\hybridengine\modeling\_utils.py
9 2 1
A
colossalai\auto_parallel\tensor_shard\node_handler\bmm_handler.py
9 5 -
A
colossalai\legacy\inference\tensor_parallel\modeling\_utils.py
9 2 1
A
examples\language\palm\palm_pytorch\autoregressive_wrapper.py
9 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\binary_elementwise_generator.py
9 3 -
A
colossalai\legacy\nn\layer\parallel_1d\_operation.py
9 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\placeholder_generator.py
9 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\where_generator.py
9 4 -
A
colossalai\legacy\nn\layer\wrapper\pipeline_wrapper.py
9 3 -
A
tests\test_fp8\test_fp8_ddp_comm_hook.py
9 1 -
A
examples\tutorial\sequence_parallel\model\layers\embedding.py
9 2 -