Using a TPU with a Jupyterlab notebook to run JAX code

11 views

Skip to first unread message

unread,

Sep 22, 2021, 9:16:15 AM9/22/21

to gce-discussion

Hello Community,

I wish to run JAX on a TPU v3-8 using a Jupyterlab notebook. What are the steps I should follow to be able to do so?

Reply all

Reply to author

Forward

0 new messages