I've deleted my tweet comparing different approaches to writing DL training loops because some folks told me it mad JAX look more difficult than it is. I'm still learning JAX (which has been a lot of fun), and if the JAX code I write is bad, that's entirely on me.
-
I enjoyed your tweet, and the discussion! It shone with good faith & curiosity. I am eager to see different perspectives, esp yours. (And let's not rule out the possibility that JAX is deficient here... JAX has many weaknesses. We love to hear about them, and to try to improve!)
Thanks, Matthew! I think JAX is great. It's a different take from what I'm used to, which is super interesting. It's just taking me some effort to get used to the new mental models and APIs.
