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\ColossalChat\coati\distributed\zero_bubble\requirements.txt
 1 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_rotary_embdding_unpad.py
3 2 -
A
tests\test_infer\test_kernels\cuda\test_rms_layernorm.py
1 1 -
A
tests\test_infer\test_async_engine\test_request_tracer.py
5 1 -
A
tests\test_infer\_utils.py
3 5 -
A
tests\test_fx\test_tracer\test_torchaudio_model\torchaudio_utils.py
6 7 -
A
tests\test_fx\test_tracer\test_hf_model\hf_tracer_utils.py
8 5 -
A
tests\test_fx\test_tracer\test_activation_checkpoint_annotation.py
10 4 -
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 -