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: 705221476
google · Dec 11, 2024 · cdf946e · cdf946e
1 parent 284b55f
commit cdf946e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/jax/carfac.py b/python/jax/carfac.py
@@ -2534,7 +2534,7 @@ def run_segment_jit_in_chunks_notraceable(
     RuntimeError: If this function is being JITTed, which it should not be.
   """
   # We add a check for tracer, until a superior fix is fixed. Tracked in
-  # https://github.com/google/jax/issues/18544 .
+  # https://github.com/jax-ml/jax/issues/18544 .
   if isinstance(input_waves, jax.core.Tracer) or isinstance(
       weights, jax.core.Tracer
   ):