Update references to JAX's GitHub repo

JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax

PiperOrigin-RevId: 704787056

Author: jakeharmon8

bayeux/__init__.py

@@ -19,7 +19,7 @@
 __version__ = '0.1.15'
 
 # Note: import <name> as <name> is required for names to be exported.
-# See PEP 484 & https://github.com/google/jax/issues/7570
+# See PEP 484 & https://github.com/jax-ml/jax/issues/7570
 # pylint: disable=useless-import-alias
 from bayeux import mcmc as mcmc
 from bayeux import optimize as optimize