jax

872a43d7 - Clarify jax.make_jaxpr docstring