main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\nn\parallel\layers\embedding.py
5 3 -
A
colossalai\legacy\nn\parallel\layers\linear.py
5 3 -
A
tests\test_lazy\test_from_pretrained.py
5 1 -
A
colossalai\booster\accelerator.py
5 3 -
A
tests\test_infer\test_request_handler.py
5 1 -
A
examples\language\gpt\experiments\auto_offload\model_zoo.py
5 2 -
A
tests\test_fp8\test_fp8_reduce_scatter.py
5 2 -
A
tests\test_infer\test_kernels\triton\test_xine_copy.py
5 1 -
A
colossalai\shardformer\shard\utils.py
5 1 -
A
applications\Colossal-LLaMA\inference\stream_chat_example.py
5 1 -
A
applications\ColossalEval\setup.py
5 2 -
A
tests\test_fp8\test_fp8_hook.py
5 1 -
A
extensions\pybind\softmax\scaled_upper_triangle_masked_softmax_cuda.py
5 1 -
A
tests\test_fx\test_graph_manipulation.py
5 3 -
A
extensions\pybind\optimizer\fused_optimizer_cuda.py
5 1 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
colossalai\fx\profiler\experimental\profiler_module\attention.py
5 2 -
A
extensions\pybind\layernorm\layernorm_cuda.py
6 1 -
A
tests\test_fp8\test_fp8_all_to_all.py
6 3 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_function\bias_addition_function.py
6 4 -