def new_jaxpr():
    jaxpr = api.make_jaxpr(scan_fun)(c, xs).jaxpr
    scan = next((eqn for eqn in jaxpr.eqns if (eqn.primitive.name == 'scan')))
    return (jaxpr, scan)
def new_jaxpr():
    jaxpr = api.make_jaxpr(scan_fun)(c, xs).jaxpr
    scan = next((eqn for eqn in jaxpr.eqns if (eqn.primitive.name == 'scan')))
    return (jaxpr, scan)