1SAA adf7e50325 Added TPExpert for special situation

2022-02-28 13:55:35 +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-02-28 13:55:35 +08:00

parallel_1d

fixed padding index issue for vocab parallel embedding layers; updated 3D linear to be compatible with examples in the tutorial

2022-02-22 02:05:43 +00:00

parallel_2d

fixed padding index issue for vocab parallel embedding layers; updated 3D linear to be compatible with examples in the tutorial

2022-02-22 02:05:43 +00:00

parallel_2p5d

fixed padding index issue for vocab parallel embedding layers; updated 3D linear to be compatible with examples in the tutorial

2022-02-22 02:05:43 +00:00

parallel_3d

fixed padding index issue for vocab parallel embedding layers; updated 3D linear to be compatible with examples in the tutorial

2022-02-22 02:05:43 +00:00

parallel_sequence

adapted for sequence parallel (#163)

2022-01-20 13:44:51 +08:00

utils

moved env variables to global variables; (#215)

2022-02-15 11:31:13 +08:00

vanilla

moved env variables to global variables; (#215)

2022-02-15 11:31:13 +08:00

wrapper

Fixed docstring in colossalai (#171)

2022-01-21 10:44:30 +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