def _custom_vjp_call_jaxpr(*args: TfVal, fun_jaxpr: core.ClosedJaxpr, **_) -> Sequence[TfVal]:
    return _interpret_jaxpr(fun_jaxpr, *args)
def _custom_vjp_call_jaxpr(*args: TfVal, fun_jaxpr: core.ClosedJaxpr, **_) -> Sequence[TfVal]:
    return _interpret_jaxpr(fun_jaxpr, *args)