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/nn/layer
BoxiangW 4a3d3446b0
Update layer integration documentations (#108)
3 years ago
..
colossalai_layer Update layer integration documentations (#108) 3 years ago
moe Added MoE parallel (#127) 3 years ago
parallel_1d Update layer integration documentations (#108) 3 years ago
parallel_2d Update layer integration documentations (#108) 3 years ago
parallel_2p5d Update layer integration documentations (#108) 3 years ago
parallel_3d Update layer integration documentations (#108) 3 years ago
parallel_sequence Migrated project 3 years ago
utils Hotfix/Colossalai layers (#92) 3 years ago
vanilla Update layer integration documentations (#108) 3 years ago
wrapper fix layers/schedule for hybrid parallelization (#111) (#112) 3 years ago
__init__.py Hotfix/Colossalai layers (#92) 3 years ago
base_layer.py Migrated project 3 years ago