def testIssue883(self):

    @partial(api.jit, static_argnums=(1,))
    def f(x, v):
        return x
    x = jnp.ones((10, 10))
    v = jnp.array([1, 2, 3])
    first_call = f(x, v)
    second_call = f(x, v)
def testIssue883(self):

    @partial(api.jit, static_argnums=(1,))
    def f(x, v):
        return x
    x = jnp.ones((10, 10))
    v = jnp.array([1, 2, 3])
    first_call = f(x, v)
    second_call = f(x, v)