def f(c, a):
    b = jnp.cos(((jnp.sum(a) + jnp.sum(c)) + jnp.sum(d)))
    c = jnp.sin((c * b))
    return (c, b)
def f(c, a):
    b = jnp.cos(((jnp.sum(a) + jnp.sum(c)) + jnp.sum(d)))
    c = jnp.sin((c * b))
    return (c, b)