def sqrt(x):
    """ Calculate the square root of the input AD object, integer, or float
    
    INPUTS
    =======
    x: input value, an AD object, int, or float
    
    RETURNS
    ========
    result: square root of x
    
    EXAMPLES
    =========
    >>> x = ad.AD(2.0, [1.0,0.0])
    >>> y = ad.AD(3.0, [0.0,1.0])
    >>> print(sqrt(x*y))
    AD(2.449489742783178, [0.61237244 0.40824829])
    """
    return x**(1/2)