jax.extend.core.jaxpr_as_fun# jax.extend.core.jaxpr_as_fun = functools.partial(<class 'functools.partial'>, <function jaxpr_as_fun>)[源代码]# 参数: closed_jaxpr (ClosedJaxpr)
