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: 702887146

Author: jakeharmon8

bayeux/__init__.py

@@ -19,7 +19,7 @@
 __version__ = '0.1.14'
 
 # 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