An optical shuffle system includes a plurality of sources that are to output respective beams of light and a plurality of receivers that are to receive respective beams of light, wherein the plurality of receivers are spaced apart from the plurality of sources. The optical shuffle system further includes an output lens formed of a plurality of output sub-wavelength grating (SWG) sections, wherein each of the plurality of output SWG sections is positioned in a respective output optical path of the plurality of sources, and wherein each of the plurality of output SWG sections is to collimate and direct light received from respective ones of the plurality of sources toward respective ones of the plurality of receivers.