main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\blip2.py
20 4 -
A
applications\ColossalChat\coati\ray\lora_constructor.py
20 1 -
A
tests\test_checkpoint_io\test_gemini_torch_compability.py
20 7 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_conv_handler.py
21 3 -
A
colossalai\zero\gemini\memory_tracer\memory_stats.py
21 4 -
A
colossalai\zero\gemini\memory_tracer\static_memstats_collector.py
21 2 -
A
examples\community\roberta\pretraining\pretrain_utils.py
21 3 -
A
tests\test_legacy\test_pipeline\rpc_test_utils.py
21 2 -
A
colossalai\inference\modeling\models\stablediffusion3.py
21 2 -
A
examples\images\vit\vit_benchmark.py
21 6 -
A
examples\language\palm\palm_pytorch\palm_pytorch.py
21 5 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_split_handler.py
21 3 -
A
examples\tutorial\new_api\cifar_vit\train.py
21 5 -
A
colossalai\auto_parallel\meta_profiler\shard_metainfo.py
21 2 -
A
tests\test_shardformer\test_model\test_shard_falcon.py
21 4 -
A
tests\test_shardformer\test_model\test_shard_chatglm2.py
21 9 -
A
colossalai\legacy\inference\dynamic_batching\sampling_params.py
21 1 -
A
colossalai\legacy\inference\serving\torch_serve\Colossal_Inference_Handler.py
21 2 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\loader.py
21 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\embedding_generator.py
21 2 -