main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\nn\optimizer\nvme_optimizer.py
44 4 -
A
colossalai\shardformer\policies\mixtral.py
44 5 3
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 1 -
A
examples\images\diffusion\ldm\util.py
45 3 -
A
colossalai\_analyzer\fx\node_util.py
45 4 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\spliced_and_tokenized_dataset.py
45 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\reshape_generator.py
45 7 1
A
examples\images\diffusion\ldm\models\autoencoder.py
46 7 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\util.py
46 4 -
A
colossalai\_analyzer\_subclasses\flop_tensor.py
46 4 2
A
colossalai\interface\model.py
46 3 -
A
examples\language\gpt\titans\model\pipeline_gpt1d.py
46 4 1
A
applications\ColossalChat\examples\training_scripts\train_kto.py
46 4 3
A
colossalai\legacy\nn\_ops\_utils.py
46 3 -
A
colossalai\fx\tracer\meta_patch\patched_function\torch_ops.py
47 7 -
A
colossalai\_analyzer\fx\codegen.py
47 4 1
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\conv_strategy_generator.py
47 8 2
A
colossalai\utils\tensor_detector\tensor_detector.py
48 4 -
A
examples\images\diffusion\ldm\models\diffusion\classifier.py
48 6 -
A
extensions\csrc\kernel\arm\cpu_adam_arm.cpp
48 1 2