An electronic device may have a waveguide, an input coupler that couples light into the waveguide, and an optical coupler that couples the light out of the waveguide. A medium may be layered onto the waveguide. The input coupler may include a first surface relief grating (SRG) and the optical coupler may include a second SRG in the medium. The first SRG may be a non-binary staircase SRG having ridges with multiple steps. The second SRG may be a binary SRG having troughs that extend to the height of the lowest steps of the first SRG. A reflective layer may fill the first SRG. Forming the input coupler using a staircase SRG may serve to maximize the in-coupling efficiency of the input coupler, thereby maximizing the efficiency with which the light is provided to an eye box.