doc nit

Author: crusaderky

src/array_api_extra/_lib/_lazy.py

@@ -426,7 +426,8 @@ def lazy_raise(  # numpydoc ignore=SA04
     Notes
     -----
     This function will raise if the :doc:`jax:transfer_guard` is active and `cond` is
-    a JAX array on a non-CPU device.
+    a JAX array on a non-CPU device
+    (`jax-ml/jax#25995 <https://github.com/jax-ml/jax/issues/25998>`_).
     """
 
     def _lazy_raise(x: Array, cond: Array) -> Array:  # numpydoc ignore=PR01,RT01
@@ -523,6 +524,10 @@ def lazy_warn(  # type: ignore[no-any-explicit]  # numpydoc ignore=SA04,PR04
 
     Notes
     -----
+    This function will raise if the :doc:`jax:transfer_guard` is active and `cond` is
+    a JAX array on a non-CPU device
+    (`jax-ml/jax#25995 <https://github.com/jax-ml/jax/issues/25998>`_).
+
     On Dask, the warning is typically going to appear on the log of the
     worker executing the function instead of on the client.
     """