Yuer867 4a0f8c2c50 fix format parallel_2p5d (#357)		2022-03-11 15:50:28 +08:00
..
colossalai_layer	moved env variables to global variables; (#215)	2022-02-15 11:31:13 +08:00
moe	Added TPExpert for special situation	2022-03-11 15:50:28 +08:00
parallel_1d	Qifan formated file ColossalAI\colossalai\nn\layer\parallel_1d\layers.py (#342)	2022-03-11 15:50:28 +08:00
parallel_2d	fixed padding index issue for vocab parallel embedding layers; updated 3D linear to be compatible with examples in the tutorial	2022-03-11 15:50:28 +08:00
parallel_2p5d	fix format parallel_2p5d (#357)	2022-03-11 15:50:28 +08:00
parallel_3d	fix format for dir-[parallel_3d] (#333)	2022-03-11 15:50:28 +08:00
parallel_sequence	adapted for sequence parallel (#163)	2022-01-20 13:44:51 +08:00
utils	flake8 style (#352)	2022-03-11 15:50:28 +08:00
vanilla	flake8 style (#352)	2022-03-11 15:50:28 +08:00
wrapper	flake8 style (#352)	2022-03-11 15:50:28 +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