Yes but its not mature enough to kill Tensorflow in the short term. I still see people prefering Tensorflow over Pytorch because they have the feeling that it

Yes but its not mature enough to kill Tensorflow in the short term. I still see people prefering Tensorflow over Pytorch because they have the feeling that it is more mature for production use. Meanwhile Jax has not converged on a recommended deep neural network framework (it has the low level pieces). At the moment its a great building block that researcher should probably know.