Merge pull request jax-ml#27821 from j-towns:clarify-make-jaxpr-docstr

PiperOrigin-RevId: 745216259

Author: Google-ML-Automation

jax/_src/api.py

@@ -2224,7 +2224,7 @@ def make_jaxpr(
     return_shape: bool = False,
     abstracted_axes: Any | None = None,
 ) -> Callable[..., core.ClosedJaxpr | tuple[core.ClosedJaxpr, Any]]:
-  """Creates a function that produces its jaxpr given example args.
+  """Create a function that returns the jaxpr of ``fun`` given example args.
 
   Args:
     fun: The function whose ``jaxpr`` is to be computed. Its positional