3 ms·
JAX replaced TF by many Google research people internally, like Google Brain, and also DeepMind. However, TF is still very actively used by Google as well, esp
by albertzeyer 4y ago
JAX replaced TF by many Google research people internally, like Google Brain, and also DeepMind. However, TF is still very actively used by Google as well, esp for production. Google is big.
- NeutralForest 4y agoI just seen new interesting research using JAX like Kevin Murphy's "Probabilistic machine learning" https://probml.github.io/pml-book/ https://probml.github.io/pml-book/. Also using TF is a PITA so all in all, I don't want to spend time using it.