main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\registry\registry.py
12 2 -
A
tests\test_legacy\test_layers\test_2d\checks_2d\check_operation_2d.py
12 4 -
A
colossalai\legacy\tensor\distspec.py
12 2 -
A
tests\kit\model_zoo\registry.py
12 8 -
A
colossalai\booster\plugin\plugin_base.py
12 3 -
A
colossalai\legacy\zero\gemini\ophooks\_shard_param_ophook.py
12 2 -
A
tests\test_infer\test_streamingllm.py
12 1 -
A
colossalai\quantization\utils.py
12 1 -
A
colossalai\fx\passes\shard_1d_pass.py
12 9 -
A
colossalai\fx\profiler\experimental\profiler_module\convolution.py
12 3 -
A
applications\ColossalQA\colossalqa\retrieval_conversation_universal.py
12 4 -
A
examples\language\gpt\titans\train_gpt.py
12 5 -
A
colossalai\tensor\d_tensor\layout.py
13 9 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_diffuser.py
13 8 -
A
extensions\pybind\layernorm\layer_norm.cpp
13 1 -
A
tests\kit\model_zoo\torchrec\torchrec.py
13 5 -
A
examples\images\diffusion\ldm\data\base.py
13 3 -
A
tests\test_shardformer\test_layer\test_sequence_parallel.py
13 2 -
A
tests\test_legacy\test_utils\test_norm_gradient_clipping.py
13 4 -
A
tests\test_legacy\test_trainer\test_pipeline\test_p2p.py
13 6 -