You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/context/process_group_initializer
Frank Lee e2089c5c15
adapted for sequence parallel (#163)
3 years ago
..
__init__.py Added MoE parallel (#127) 3 years ago
initializer_1d.py Hotfix/Colossalai layers (#92) 3 years ago
initializer_2d.py Migrated project 3 years ago
initializer_2p5d.py Develop/experiments (#59) 3 years ago
initializer_3d.py Develop/experiments (#59) 3 years ago
initializer_data.py Migrated project 3 years ago
initializer_model.py Optimize pipeline schedule (#94) 3 years ago
initializer_moe.py Added MoE parallel (#127) 3 years ago
initializer_pipeline.py Optimize pipeline schedule (#94) 3 years ago
initializer_sequence.py adapted for sequence parallel (#163) 3 years ago
initializer_tensor.py Migrated project 3 years ago
process_group_initializer.py Migrated project 3 years ago