main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\auto_parallel\tensor_shard\node_handler\tensor_constructor_handler.py
2 2 -
A
colossalai\legacy\pipeline\rpc\_pipeline_base.py
280 2 -
A
examples\images\diffusion\ldm\data\imagenet.py
79 2 3
A
colossalai\shardformer\__init__.py
 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\transpose_handler.py
8 2 -
A
colossalai\kernel\triton\softmax.py
10 2 -
A
examples\images\diffusion\ldm\lr_scheduler.py
19 2 -
A
colossalai\legacy\pipeline\rpc\__init__.py
 2 -
A
examples\images\diffusion\configs\Inference\x4-upscaling.yaml
 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\view_handler.py
3 2 -
A
colossalai\shardformer\examples\convergence_benchmark.sh
 2 -
A
colossalai\legacy\nn\parallel\layers\__init__.py
 2 -
A
colossalai\legacy\builder\__init__.py
 2 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\__init__.py
 2 -
A
colossalai\auto_parallel\tensor_shard\options.py
 2 -
A
colossalai\shardformer\modeling\chatglm2_6b\configuration_chatglm.py
1 2 -
A
colossalai\shardformer\examples\data.py
23 2 -
A
colossalai\legacy\nn\parallel\layers\cache_embedding\copyer.py
5 2 -
A
colossalai\legacy\amp\naive_amp\naive_amp.py
38 2 -
A
colossalai\legacy\communication\p2p.py
54 2 -