Request edit access
JAX on Cloud TPU Interest Form
IMPORTANT UPDATE (May 26, 2021): Cloud TPU VMs are now publicly available and you do not need to fill out this form to get access!

We're glad you're interested in trying JAX with the new Cloud TPU experience! This is still a "preview" release, meaning there may still be some rough edges, but we hope you’ll give JAX a try on the new Cloud TPU experience and let us know what you think.

To get started, check out the Cloud TPU Preview documentation here: We recommend the JAX quickstart in particular:

If you need to request additional Cloud TPU quota, more information is available here If applicable, you may also be interested in signing up for the TPU Research Cloud program to receive free Cloud TPU quota:

If you run into any issues that are specific to JAX, please file an issue ( or post a discussion question ( You’re also welcome to email us directly at

If you run into general Cloud TPU issues (e.g. trouble creating a Cloud TPU VM), or you’re not sure if the issue is specific to JAX, please email, or if you are a TRC member.

We’d also love to hear any feedback you have, good or bad, at Hope to hear from you!
Sign in to Google to save your progress. Learn more
Name *
Email *
Affiliation (your university, company, etc.) *
(Optional) Tell us what you'd like to do with JAX on Cloud TPUs!
(Optional) Social links (GitHub, LinkedIn, Twitter, etc.)
Clear form
Never submit passwords through Google Forms.
This form was created inside of Privacy & Terms