jax.extend.core.jaxpr_as_fun# jax.extend.core.jaxpr_as_fun = functools.partial(<class 'functools.partial'>, <function jaxpr_as_fun>)[source]# Parameters: closed_jaxpr (ClosedJaxpr)
