main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\zero\gemini\gemini_context.py
8 2 -
A
tests\test_auto_parallel\test_offload\model_utils.py
8 3 -
A
tests\test_zero\test_low_level\test_grad_acc.py
8 6 -
A
tests\test_zero\test_gemini\test_zerooptim_state_dict.py
8 6 -
A
tests\test_zero\test_gemini\test_chunk_mgrv2.py
8 4 -
A
tests\test_analyzer\test_fx\test_shape_prop.py
8 4 -
A
examples\tutorial\opt\inference\script\process-opt-175b\convert_ckpt.py
8 2 -
A
tests\test_infer\test_cuda_graph.py
8 1 -
A
colossalai\nn\layer\layernorm.py
8 1 -
A
applications\ColossalQA\colossalqa\chain\retrieval_qa\stuff.py
8 2 -
A
applications\ColossalChat\coati\ray\detached_replay_buffer.py
8 1 -
A
tests\test_booster\test_mixed_precision\test_fp16_torch.py
7 3 -
A
colossalai\fx\tracer\bias_addition_patch\patched_bias_addition_function\addmm.py
7 3 -
A
examples\inference\client\locustfile.py
7 2 -
A
examples\tutorial\sequence_parallel\model\layers\linear.py
7 2 -
A
examples\inference\benchmark_ops\benchmark_fused_rotary_embdding_unpad.py
7 1 -
A
examples\inference\benchmark_ops\benchmark_kv_cache_memcopy.py
7 1 -
A
examples\language\model_utils.py
7 1 -
A
tests\test_fx\test_tracer\test_torchvision_model\test_torchvision_model.py
7 9 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_placeholder_handler.py
7 7 -