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.