main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\inference\flash_decoding_utils.py
8 1 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_softmax_handler.py
8 3 -
A
tests\test_auto_parallel\test_offload\test_solver.py
8 3 -
A
applications\ColossalChat\benchmarks\dummy_dataset.py
8 2 -
A
colossalai\inference\modeling\policy\nopadding_baichuan.py
8 4 -
A
colossalai\inference\modeling\policy\pixart_alpha.py
8 2 -
A
colossalai\inference\modeling\policy\stablediffusion3.py
8 2 -
A
colossalai\inference\spec\drafter.py
8 2 -
A
tests\test_auto_parallel\test_offload\model_utils.py
8 3 -
A
examples\inference\benchmark_ops\benchmark_rmsnorm.py
8 1 -
A
colossalai\legacy\context\process_group_initializer\initializer_sequence.py
8 2 -
A
colossalai\legacy\engine\gradient_handler\utils.py
8 1 -
A
setup.py
8 9 -
A
examples\tutorial\opt\inference\script\process-opt-175b\convert_ckpt.py
8 2 -
A
colossalai\legacy\inference\dynamic_batching\stats.py
8 1 -
A
tests\test_fx\test_tracer\test_hf_model\hf_tracer_utils.py
8 5 -
A
colossalai\auto_parallel\tensor_shard\node_handler\binary_elementwise_handler.py
8 7 -
A
colossalai\legacy\inference\tensor_parallel\policies\llama.py
8 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getattr_generator.py
8 3 -
A
examples\language\grok-1\grok1_policy.py
8 1 -