|
colossalai_layer
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
moe
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
parallel_1d
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
parallel_2d
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
parallel_2p5d
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
parallel_3d
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
parallel_sequence
|
adapted for sequence parallel (#163)
|
2022-01-20 13:44:51 +08:00 |
|
utils
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +08:00 |
|
vanilla
|
Fixed docstring in colossalai (#171)
|
2022-01-21 10:44:30 +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 |
0f8c7f9804