Do we need pmap
to take advantage of all 8 cores of TPUv2 on Colab?
#9055
Unanswered
NightMachinery
asked this question in
Q&A
Replies: 0 comments
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.
-
Do we need
pmap
to take advantage of all 8 cores of TPUv2 on Colab? If so, is there no difference between parallelism between TPU cores and parallelism between multiple GPUs?Beta Was this translation helpful? Give feedback.
All reactions