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.
@ikostrikov Yeah when it started out, TF was great too. I pray to give Jax devs the strength to keep a simple core and avoid bloat ;-)