def beam_curv(z, z0):
    """
    Computes the radius of the curvature of the beam wavefront at an axial distance of
    z from the beam's waist.

    Args:
        z: axial distance from beam's waist
        z0: rayleigh range
    Returns:
        radius of curvature
    """

    return z + z0**2 / z