main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\unary_elementwise_generator.py
13 4 -
A
applications\ColossalChat\coati\experience_buffer\naive.py
13 2 -
A
extensions\pybind\layernorm\layer_norm.cpp
13 1 -
A
applications\ColossalEval\colossal_eval\dataset\gsm.py
13 2 -
A
examples\language\data_utils.py
13 2 -
A
colossalai\legacy\nn\layer\parallel_sequence\layers.py
13 4 -
A
colossalai\fx\profiler\experimental\profiler_function\arithmetic.py
13 3 -
A
tests\test_legacy\test_trainer\test_pipeline\test_p2p.py
13 6 -
A
applications\ColossalQA\colossalqa\mylogging.py
13 2 -
A
colossalai\fx\tracer\_tracer_utils.py
12 7 -
A
colossalai\quantization\utils.py
12 1 -
A
colossalai\fx\tracer\meta_patch\patched_function\convolution.py
12 3 -
A
colossalai\fx\tracer\meta_patch\patched_module\pooling.py
12 4 -
A
tests\test_legacy\test_layers\test_2d\checks_2d\check_operation_2d.py
12 4 -
A
tests\test_legacy\test_layers\test_2p5d\checks_2p5d\check_operation_2p5d.py
12 4 -
A
colossalai\legacy\inference\pipeline\benchmark\benchmark.py
12 2 -
A
colossalai\legacy\zero\gemini\ophooks\_shard_param_ophook.py
12 2 -
A
colossalai\auto_parallel\passes\comm_metainfo_pass.py
12 3 -
A
applications\ColossalChat\coati\distributed\loss.py
12 1 -
A
tests\test_shardformer\test_layer\test_ring_attn.py
12 2 -