Very much alive. From what I can tell it has more or less replaced Tensorflow for research purposes. (A lot of researchers use PyTorch though.)

Development seems not to have dropped at all from the contributions page: https://github.com/google/jax/graphs/contributors

Don’t know about usage and uptake though.

Why would it be dead?