main
ColossalAI
Repository badge with A rating
View on GitHub
Grade
Name
Complexity
Churn
Issues
A
colossalai\checkpoint_io\checkpoint_io_base.py
32 8 -
A
tests\test_pipeline\test_p2p_communication.py
17 8 -
A
tests\test_lazy\lazy_init_utils.py
15 8 -
A
colossalai\inference\__init__.py
 8 -
A
examples\images\diffusion\configs\train_colossalai_cifar10.yaml
 8 -
A
colossalai\fx\passes\utils.py
27 8 -
A
tests\test_fx\test_meta_info_prop.py
6 8 -
A
colossalai\shardformer\layer\loss.py
25 8 -
A
colossalai\fx\passes\split_module.py
35 8 -
A
tests\test_booster\test_plugin\test_torch_ddp_plugin.py
19 8 -
A
tests\test_fx\test_coloproxy.py
3 8 -
A
colossalai\shardformer\layer\linear.py
85 8 -
A
colossalai\auto_parallel\tensor_shard\node_handler\strategy\conv_strategy_generator.py
47 8 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_output_handler.py
6 8 -
A
colossalai\fx\__init__.py
 8 -
A
colossalai\utils\model\utils.py
18 8 -
A
colossalai\nn\optimizer\hybrid_adam.py
17 8 -
A
tests\test_fx\test_tracer\test_hf_model\test_hf_diffuser.py
13 8 -
A
colossalai\checkpoint_io\moe_checkpoint.py
155 8 -
A
tests\test_auto_parallel\test_tensor_shard\test_node_handler\test_norm_pooling_handler.py
3 8 -