A system for an extreme ultraviolet (EUV) light source includes a light-generation system configured to emit one or more light beams onto a beam path; one or more optical amplifiers, each of the one or more amplifiers including a gain medium on the beam path, each gain medium being configured to amplify the one or more light beams to produce one or more amplified light beams; and one or more diffractive optical elements on the beam path, where each of the one or more diffractive optical elements has a plurality of focal lengths, and each focal length of the diffractive optical element is associated with a particular polarization state.