main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
examples\language\gpt\hybridparallelism\data.py
20 3 -
A
examples\inference\llama\benchmark_llama3.py
20 2 -
A
colossalai\legacy\zero\gemini\ophooks\utils.py
20 2 -
A
applications\ColossalChat\coati\ray\lora_constructor.py
20 1 -
A
colossalai\context\config.py
20 2 -
A
colossalai\inference\core\plugin.py
20 2 -
A
examples\images\resnet\train.py
20 4 -
A
examples\tutorial\new_api\cifar_resnet\train.py
20 5 -
A
colossalai\zero\gemini\utils.py
20 4 -
A
tests\test_shardformer\test_model\test_shard_vit.py
20 5 -
A
tests\test_shardformer\test_model\test_shard_opt.py
20 4 -
A
tests\test_shardformer\test_model\test_shard_gptj.py
20 4 -
A
tests\test_shardformer\test_model\test_shard_bloom.py
20 6 -
A
tests\test_checkpoint_io\test_gemini_torch_compability.py
20 7 -
A
colossalai\zero\gemini\memory_tracer\memory_monitor.py
19 4 -
A
examples\images\diffusion\ldm\modules\ema.py
19 3 -
A
colossalai\zero\gemini\memory_tracer\runtime_mem_tracer.py
19 5 -
A
colossalai\nn\optimizer\lars.py
19 4 -
A
examples\language\gpt\experiments\pipeline_parallel\train_gpt_pp.py
19 7 -
A
examples\language\bert\data.py
19 4 -