main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\legacy\inference\tensor_parallel\engine.py
52 1 -
A
colossalai\legacy\trainer\_trainer.py
51 4 -
A
colossalai\legacy\inference\manager.py
51 2 -
A
applications\ColossalChat\coati\dataset\tokenization_utils.py
51 6 -
A
colossalai\legacy\utils\checkpointing.py
51 2 -
A
colossalai\fx\passes\passes_for_gpt2_test.py
50 6 -
A
colossalai\accelerator\base_accelerator.py
50 1 -
A
colossalai\shardformer\policies\command.py
50 4 -
A
tests\test_optimizer\test_dist_adafactor.py
50 4 -
A
examples\tutorial\sequence_parallel\data\tokenizer\tokenizer.py
50 3 -
A
colossalai\shardformer\policies\opt.py
50 13 -
A
colossalai\auto_parallel\tensor_shard\solver\cost_graph.py
49 7 -
A
colossalai\shardformer\policies\mistral.py
49 4 -
A
applications\ColossalChat\benchmarks\benchmark_ppo.py
49 2 -
A
extensions\csrc\kernel\arm\cpu_adam_arm.cpp
48 1 -
A
examples\images\diffusion\ldm\models\diffusion\classifier.py
48 6 -
A
colossalai\lazy\pretrained.py
48 3 -
A
colossalai\utils\tensor_detector\tensor_detector.py
48 4 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\conv_strategy_generator.py
47 8 -
A
colossalai\shardformer\policies\gptj.py
47 11 -