I'm not aware of any "actively pushing JAX" agenda. Of course library developers are always happy to present their work. But all jax users I'm aware of switched just because it's a beautifully crafted library and a joy to use, esp. with TPUs! It's as simple as that.
I'm not aware of any "actively pushing JAX" agenda. Of course library developers are always happy to present their work. But all jax users I'm aware of switched just because it's a beautifully crafted library and a joy to use, esp. with TPUs! It's as simple as that.
@giffmana I was a Jax skeptic myself until I tried it. Also, I guess it's fair to say that every new library might be better than the previous ones.
@giffmana Know of any good Torch vs Jax commentary? Curious to dive in, but also want a sort of high level pro/con if one exists
@giffmana If "pushing" was enough, TF would not have lost so much momentum.
@giffmana JAX growth has been very organic, long term this might be better since the JAX team will have more time to polish the APIs, TF's meteoric rise didn't end so well.
@giffmana JAX was easy enough to get a hang of that I didn't drop it immediately, and also fit my needs perfectly (ML/differential equations) so it stuck. I've shared it with people and most responses are "This is so cool!" The organic growth is real.