main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
applications\ColossalQA\examples\retrieval_conversation_universal.py
 3 -
A
examples\language\bert\benchmark.py
16 3 -
A
examples\images\diffusion\scripts\txt2img.sh
 3 -
A
applications\ColossalQA\examples\retrieval_conversation_zh.py
 3 -
A
examples\images\vit\run_demo.sh
 3 -
A
colossalai\fx\profiler\memory_utils.py
16 3 -
A
tests\kit\model_zoo\transformers\gptj.py
4 3 -
A
examples\language\grok-1\run_inference_slow.sh
 3 -
A
examples\images\vit\run_benchmark.sh
 3 -
A
tests\test_analyzer\test_fx\test_nested_ckpt.py
10 3 -
A
tests\kit\model_zoo\transformers\mistral.py
3 3 -
A
examples\tutorial\auto_parallel\auto_ckpt_batchsize_test.py
4 3 -
A
colossalai\shardformer\shard\__init__.py
 3 -
A
colossalai\shardformer\shard\grad_ckpt_config.py
11 3 -
A
colossalai\fx\tracer\_symbolic_trace.py
2 3 -
A
tests\kit\model_zoo\torchrec\__init__.py
 3 -
A
.github\workflows\scripts\example_checks\check_dispatch_inputs.py
5 3 -
A
colossalai\tensor\comm_spec.py
61 3 -
A
applications\ColossalChat\coati\trainer\rm.py
20 3 -
A
examples\language\gpt\gemini\commons\model_zoo.py
20 3 -