main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
tests\test_infer\test_cuda_graph.py
8 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getattr_generator.py
8 3 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\tensor_constructor_generator.py
8 2 -
A
colossalai\legacy\nn\layer\parallel_sequence\_operation.py
8 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\transpose_handler.py
8 2 -
A
colossalai\legacy\nn\loss\loss_2d.py
8 4 -
A
colossalai\legacy\nn\loss\loss_2p5d.py
8 4 -
A
colossalai\legacy\nn\loss\loss_3d.py
8 4 -
A
tests\test_auto_parallel\test_offload\model_utils.py
8 3 -
A
applications\ColossalChat\coati\experience_maker\base.py
8 1 -
A
examples\tutorial\auto_parallel\auto_parallel_with_resnet.py
8 4 -
A
colossalai\legacy\zero\gemini\gemini_context.py
8 2 -
A
applications\Colossal-LLaMA\colossal_llama\utils\utils.py
8 1 -
A
tests\test_auto_parallel\test_offload\test_perf.py
8 5 -
A
tests\test_analyzer\test_fx\test_shape_prop.py
8 4 -
A
colossalai\nn\layer\layernorm.py
8 1 -
A
examples\language\grok-1\grok1_policy.py
8 1 -
A
colossalai\pipeline\weight_grad_store.py
8 2 -
A
colossalai\shardformer\layer\dropout.py
8 10 -
A
applications\ColossalQA\colossalqa\chain\retrieval_qa\stuff.py
8 2 -