How to realize model-paralllel using jax #8167
Unanswered
DahengWang
asked this question in
Q&A
Replies: 1 comment
-
|
Any insights since this question was published? |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
I'm trying to build a model using jax under model-parallel and data-parallel. It seems that data-parallel can be realized by pmap, but I falied to find how to realize model-parallel.
Beta Was this translation helpful? Give feedback.
All reactions