main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_checkpoint_io\test_gemini_torch_compability.py
20 7 -
A
extensions\csrc\kernel\cuda\multi_tensor_apply.cuh
 1 -
A
tests\test_booster\test_plugin\test_3d_plugin.py
29 3 -
A
tests\test_auto_parallel\test_offload\test_solver.py
8 3 -
A
tests\test_booster\test_plugin\test_dp_plugin_base.py
16 4 -
A
extensions\csrc\kernel\cuda\multi_tensor_sgd_kernel.cu
 1 -
A
colossalai\auto_parallel\offload\util.py
22 3 -
A
.github\workflows\release_docker_after_publish.yml
 5 -
A
tests\test_booster\test_plugin\test_gemini_plugin.py
24 2 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_bert.py
3 13 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_albert.py
3 3 -
A
tests\test_autochunk\test_autochunk_alphafold\test_autochunk_alphafold_utils.py
22 4 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_norm_pooling_handler.py
3 8 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_batch_norm_handler.py
4 3 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_conv_metainfo.py
6 3 -
A
tests\kit\model_zoo\transformers\mixtral.py
3 4 -
A
extensions\pybind\optimizer\fused_optimizer_cuda.py
5 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 -