diff --git a/clu/profiler.py b/clu/profiler.py
index f946f8a..86af11c 100644
--- a/clu/profiler.py
+++ b/clu/profiler.py
@@ -31,8 +31,9 @@ def start(logdir: str, options=None):
   if options is not None:
     raise NotImplementedError(
         "'options' not supported by clu.profiler.start(). Please file an issue "
-        "at https://github.com/google/jax/issues requesting profiler option "
-        "support if you need this feature.")
+        "at https://github.com/jax-ml/jax/issues requesting profiler option "
+        "support if you need this feature."
+    )
   if logdir is None:
     raise ValueError("Must specify logdir where profile should be written!")
   jax.profiler.start_trace(logdir)
diff --git a/clu_synopsis.ipynb b/clu_synopsis.ipynb
index 97a9008..68d2ad1 100644
--- a/clu_synopsis.ipynb
+++ b/clu_synopsis.ipynb
@@ -16,7 +16,7 @@
         "https://github.com/google/CommonLoopUtils\n",
         "\n",
         "This package is usually used with\n",
-        "[JAX](https://github.com/google/jax)\n",
+        "[JAX](https://github.com/jax-ml/jax)\n",
         "/\n",
         "[Flax](https://github.com/google/flax)\n",
         "ML projects, but it can also be used with other ML frameworks.\n",