5 ms·
TF is being abandoned for JAX in many cases, most of the new things I see coming up in ML are done in JAX an Pytorch stays strong. It doesn't help that the TF d
by NeutralForest 4y ago
TF is being abandoned for JAX in many cases, most of the new things I see coming up in ML are done in JAX an Pytorch stays strong. It doesn't help that the TF docs are a catastrophe.
- GistNoesis 4y agoAll you need to know about JAX from JAX main github page : "This is a research project, not an official Google product. Expect bugs and sharp edges. Please help by trying it out, reporting bugs, and letting us know what you think!"
- NeutralForest 4y agoYeah it is for now but I'm not holding my breath for Google based projects and I can't see myself using TF in the future either.
- albertzeyer 4y agoJAX 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.