main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalChat\visualization.py
 1 -
A
tests\test_fp8\test_fp8_fsdp_comm_hook.py
14 2 -
A
colossalai\legacy\nn\loss\loss_2d.py
8 4 -
A
tests\test_fp8\test_fp8_linear.py
4 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\tensor_constructor_handler.py
2 2 -
A
tests\test_fx\test_codegen\test_activation_checkpoint_codegen.py
18 4 -
A
colossalai\legacy\nn\loss\loss_3d.py
8 4 -
A
tests\test_fx\test_codegen\test_offload_codegen.py
22 4 -
A
tests\test_moe\test_moe_ep_zero.py
8 3 -
A
tests\test_zero\test_low_level\test_zero1_2.py
25 3 -
A
tests\test_zero\test_low_level\test_mem_leak.py
6 1 -
A
tests\test_optimizer\test_adam_optim.py
10 2 -
A
tests\test_legacy\test_utils\test_checkpoint\test_checkpoint_2p5d.py
12 3 -
A
tests\test_legacy\test_pipeline\test_pipelinable.py
5 2 -
A
tests\test_legacy\test_moe\moe_utils.py
28 1 -
A
tests\test_infer\test_streamingllm.py
12 1 -
A
tests\test_fx\test_tracer\test_functional_conv.py
1 3 -
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 -