def bodyfun_1(inp):
    (val, i, j) = inp
    j = 0

    def bodyfun_2(inp):
        (val, i, j) = inp
        val += (i + j)
        j += 1
        return (val, i, j)
    result = lax.while_loop(condfun_2, bodyfun_2, (val, i, j))
    val = result[0]
    i += 1
    return (val, i, j)
def bodyfun_1(inp):
    (val, i, j) = inp
    j = 0

    def bodyfun_2(inp):
        (val, i, j) = inp
        val += (i + j)
        j += 1
        return (val, i, j)
    result = lax.while_loop(condfun_2, bodyfun_2, (val, i, j))
    val = result[0]
    i += 1
    return (val, i, j)