def METHOD_NAME(a, b):
    r = np.empty_like(a)
    for i in nb.prange(len(r)):
        if a[i] > b[i]:
            r[i] = a[i] + np.log(1 + np.exp(b[i] - a[i]))
        else:
            r[i] = b[i] + np.log(1 + np.exp(a[i] - b[i]))
    return r