We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bcd4048 commit bc11a63Copy full SHA for bc11a63
jax/_src/api.py
@@ -2142,7 +2142,7 @@ def make_jaxpr(
2142
return_shape: bool = False,
2143
abstracted_axes: Any | None = None,
2144
) -> Callable[..., core.ClosedJaxpr | tuple[core.ClosedJaxpr, Any]]:
2145
- """Creates a function that produces its jaxpr given example args.
+ """Create a function that returns the jaxpr of ``fun`` given example args.
2146
2147
Args:
2148
fun: The function whose ``jaxpr`` is to be computed. Its positional
0 commit comments