main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\gptj.py
47 11 -
A
colossalai\_analyzer\fx\codegen.py
47 4 -
A
tests\test_optimizer\test_dist_came.py
47 4 -
A
colossalai\legacy\nn\_ops\_utils.py
46 3 -
A
applications\ColossalChat\examples\training_scripts\train_kto.py
46 4 -
A
colossalai\booster\plugin\torch_fsdp_plugin.py
46 5 -
A
colossalai\_analyzer\_subclasses\flop_tensor.py
46 4 -
A
examples\language\gpt\titans\model\pipeline_gpt1d.py
46 4 -
A
examples\images\diffusion\ldm\modules\diffusionmodules\util.py
46 4 -
A
examples\images\diffusion\ldm\models\autoencoder.py
46 7 -
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 1 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\reshape_generator.py
45 7 -
A
colossalai\autochunk\select_chunk.py
44 5 -
A
colossalai\nn\optimizer\nvme_optimizer.py
44 4 -
A
colossalai\legacy\context\process_group_initializer\initializer_3d.py
44 2 -
A
tests\test_optimizer\_utils.py
44 3 -
A
applications\ColossalEval\colossal_eval\utils\conversation.py
44 4 -
A
applications\ColossalChat\coati\ray\detached_trainer_base.py
44 1 -