main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\gpt2.py
81 6 -
A
colossalai\tensor\d_tensor\api.py
33 6 -
A
colossalai\fx\tracer\_tracer_utils.py
12 7 -
A
colossalai\testing\__init__.py
 7 -
A
colossalai\zero\gemini\chunk\search_utils.py
25 7 -
A
colossalai\fx\tracer\meta_patch\patched_function\torch_ops.py
47 7 -
A
colossalai\zero\gemini\placement_policy.py
58 7 -
A
colossalai\inference\core\engine.py
14 7 -
A
examples\images\diffusion\configs\Teyvat\train_colossalai_teyvat.yaml
 7 -
A
examples\images\diffusion\ldm\models\autoencoder.py
46 7 -
A
examples\images\diffusion\scripts\img2img.py
23 7 -
A
examples\images\diffusion\scripts\txt2img.py
28 7 -
A
examples\language\gpt\experiments\pipeline_parallel\train_gpt_pp.py
19 7 -
A
examples\tutorial\auto_parallel\bench_utils.py
18 7 -
A
applications\ColossalChat\examples\training_scripts\train_rm.py
40 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\binary_elementwise_handler.py
8 7 -
A
tests\test_auto_parallel\test_tensor_shard\test_gpt\test_solver_with_gpt_module.py
8 7 -
A
tests\test_auto_parallel\test_tensor_shard\test_metainfo\test_activation_metainfo.py
1 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\getitem_generator.py
27 7 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\layer_norm_generator.py
26 7 -