4 ms·
> "It took me a while to realize it, but Jax is actually a huge opportunity for a lot of scientific computing." In all conferences like NeurIPS, in Google ML C
by __rito__ 3y ago
> "It took me a while to realize it, but Jax is actually a huge opportunity for a lot of scientific computing."
In all conferences like NeurIPS, in Google ML Community days, etc., whenever there is a JAX workshop/tutorial/talk, it is always touted as a numerical computation library. And it was developed as such. Sure the focus is in ML, but everyone involved in it always have said that this is a general purpose scientific computing library.
Flax, Haiku, etc. are Deep Learning libraries.
- matrss 3y agoMeanwhile, the first sentence in their readme is this: > JAX is Autograd and XLA, brought together for high-performance machine learning research. That does not really convey the generality of it that well.
- mattjjatgoogle 3y agoYou're right! Maybe we should revise that... I made https://github.com/google/jax/pull/17851 https://github.com/google/jax/pull/17851, comments welcome!
- londons_explore 3y agoSo was tensorflow... And yet it's pretty much dead.