main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\shardformer\__init__.py
 2 -
A
colossalai\legacy\nn\layer\wrapper\__init__.py
 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\permute_handler.py
12 2 -
A
applications\ColossalChat\tests\generate_dummy_datasets_for_testing.py
 2 -
A
colossalai\legacy\amp\apex_amp\__init__.py
1 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\softmax_handler.py
4 2 -
A
colossalai\pipeline\schedule\zero_bubble_pp.py
128 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\split_handler.py
6 2 -
A
colossalai\pipeline\schedule\__init__.py
 2 -
A
applications\ColossalChat\tests\test_data_preparation.sh
 2 -
A
colossalai\pipeline\stage_manager.py
33 2 -
A
applications\Colossal-LLaMA\colossal_llama\dataset\spliced_and_tokenized_dataset.py
45 2 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\embedding_generator.py
21 2 -
A
colossalai\legacy\tensor\dist_spec_mgr.py
26 2 -
A
colossalai\nn\optimizer\galore.py
61 2 -
A
colossalai\legacy\context\process_group_initializer\__init__.py
 2 -
A
examples\images\dreambooth\inference.py
 2 -
A
colossalai\utils\__init__.py
 2 -
A
colossalai\shardformer\examples\convergence_benchmark.sh
 2 -
A
colossalai\pipeline\weight_grad_store.py
8 2 -