ColossalAI/colossalai/nn/layer
Frank Lee e2089c5c15
adapted for sequence parallel (#163)
2022-01-20 13:44:51 +08:00
..
colossalai_layer Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
moe Added MoE parallel (#127) 2022-01-07 15:08:36 +08:00
parallel_1d Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
parallel_2d Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
parallel_2p5d Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
parallel_3d Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
parallel_sequence adapted for sequence parallel (#163) 2022-01-20 13:44:51 +08:00
utils Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
vanilla Update layer integration documentations (#108) 2022-01-10 18:05:58 +08:00
wrapper fix layers/schedule for hybrid parallelization (#111) (#112) 2022-01-04 20:52:31 +08:00
__init__.py Hotfix/Colossalai layers (#92) 2021-12-29 23:32:10 +08:00
base_layer.py Migrated project 2021-10-28 18:21:23 +02:00