import uarray as ua
import unumpy as np
import udiff
from unumpy import numpy_backend

with ua.set_backend(numpy_backend), ua.set_backend(udiff, coerce=True):
    x = np.reshape(np.arange(25), (5, 5))
    x.var = udiff.Variable('x')
    x.diffs[x].diffs[x] = 0

    y = np.exp(2 * x)
    print(y)
    print(y.diffs[x])
    print(y.diffs[x].diffs[x])