main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\where_handler.py
6 4 -
A
tests\test_fx\test_pipeline\test_hf_model\test_t5.py
3 7 -
A
colossalai\legacy\nn\parallel\layers\__init__.py
 2 -
A
tests\test_fx\test_pipeline\test_timm_model\timm_utils.py
4 2 -
A
applications\Colossal-LLaMA\colossal_llama\utils\ckpt_io.py
6 2 -
A
tests\test_fx\test_pipeline\test_topo\topo_utils.py
18 3 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\base_embedding.py
4 2 -
A
tests\test_fx\test_pipeline_passes.py
4 6 -
A
colossalai\auto_parallel\tensor_shard\sharding_strategy.py
32 13 -
A
tests\test_infer\test_kernels\cuda\test_flash_decoding_attention.py
16 3 -
A
tests\test_infer\test_kernels\cuda\test_convert_fp8.py
1 1 -
A
tests\test_fx\test_tracer\test_torchaudio_model\torchaudio_utils.py
6 7 -
A
tests\test_fx\test_tracer\test_timm_model\test_timm_model.py
9 17 -
A
.github\workflows\doc_test_on_pr.yml
 5 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
applications\Colossal-LLaMA\train_sft.example.sh
 1 -
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 1 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -