Use a regular import to add jax.__version__ rather than exec() trickery.
Use a regular import to add jax.__version__ rather than exec() trickery.

(The exec() trickery is needed for setup.py, but not for jax/__init__.py.)

tensorflow/probability,google/jax,tensorflow/probability,google/jax,google/jax,google/jax