main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\policies\sam.py
9 4 -
A
colossalai\fx\profiler\experimental\profiler_module\rnn.py
10 2 -
A
colossalai\shardformer\policies\t5.py
98 3 -
A
colossalai\shardformer\policies\vit.py
43 3 -
A
applications\ColossalQA\colossalqa\retrieval_conversation_zh.py
 3 -
A
colossalai\fx\profiler\experimental\profiler_module\torch_op.py
1 2 -
A
colossalai\shardformer\policies\whisper.py
88 3 -
A
colossalai\shardformer\shard\__init__.py
 3 -
A
colossalai\fx\profiler\experimental\registry.py
4 2 -
A
colossalai\shardformer\shard\grad_ckpt_config.py
11 3 -
A
colossalai\shardformer\shard\shard_config.py
14 8 -
A
applications\ColossalChat\coati\trainer\__init__.py
 2 -
A
applications\ColossalQA\colossalqa\retriever.py
27 5 -
A
colossalai\fx\profiler\experimental\shard_utils.py
3 2 -
A
colossalai\shardformer\shard\sharder.py
39 7 -
A
colossalai\shardformer\shard\shardformer.py
3 5 -
A
colossalai\fx\profiler\memory_utils.py
16 3 -
A
colossalai\shardformer\shard\utils.py
5 1 -
A
colossalai\tensor\__init__.py
 25 -
A
applications\ColossalQA\colossalqa\text_splitter\__init__.py
 1 -