from autograd import grad
from autograd import numpy as np


def f(u):
    return np.cos(u)


g = grad(f)

x = np.array(0.)
y = f(x)
z = g(x)

print(x, y, z)