Shawn Presser ⠕ on Twitter: "How to tame your Jax: Navigating TPU VMs with TFRC in 2021 https://t.co/QNp6pBtDkp"
You can add location information to your Tweets, such as your city or precise location, from the web and via third-party applications. You always have the option to delete your Tweet location history. Learn more
By embedding Twitter content in your website or app, you are agreeing to the Twitter Developer Agreement and Developer Policy.
The fastest way to share someone else’s Tweet with your followers is with a Retweet. Tap the icon to send it instantly.
How to tame your Jax: Navigating TPU VMs with TFRC in 2021 https:// github.com/google/jax/iss ues/2108#issuecomment-866238579 …
I felt a little scared too at first. And I like Pytorch so much that I even tried to port Pytorch to Tensorflow ( https:// mobile.twitter.com/theshawwn/stat us/1311925180126511104 … ) just so that, since I was forced to use tensorflow, I could still use pytorch. But jax isn't so scary. Do you know numpy? Then you know jax.
Nice! A lot of people will find this useful. Have you thought about writing a blog post on the topic? Maybe including a small (but computation-heavy) application, with a link to a GitHub repository/gist.
