ColossalAI/colossalai/nn/layer
Liang Bowen 7eb87f516d flake8 style (#352) 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 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_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