Field curvature of an optical system is modified based on topography of the surface of a wafer such that an image of each of the segments of the surface is in focus across the segment. The wafer may be non-planar. The optical system may be a multi-element lens system connected to a controller that modifies the field curvature by changing position of the lens elements. The wafer may be held by a chuck, such as an edge grip chuck. Multiple optical systems may be arranged across a dimension of the wafer.