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
jax-ml · Dec 5, 2024 · 068eec5 · 068eec5
1 parent 7fc5a2d
commit 068eec5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/bayeux/__init__.py b/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

diff --git a/docs/examples/oryx_and_bayeux.ipynb b/docs/examples/oryx_and_bayeux.ipynb