def testIssue1316(self):

    def f(carry, _):
        (c, key) = carry
        (key, _) = random.split(key)
        return ((c, key), ())
    key = random.PRNGKey(0)
    api.grad((lambda c: lax.scan(f, (c, key), np.ones(3))[0][0]))(0.0)
def testIssue1316(self):

    def f(carry, _):
        (c, key) = carry
        (key, _) = random.split(key)
        return ((c, key), ())
    key = random.PRNGKey(0)
    api.grad((lambda c: lax.scan(f, (c, key), np.ones(3))[0][0]))(0.0)