Described herein are photonic integrated circuits (PICs) comprising a semiconductor optical amplifier (SOA) to output a signal comprising a plurality of wavelengths, a sensor to detect data associated with a power value of each wavelength of the output signal of the SOA, a filter to filter power values of one or more of the wavelengths of the output signal of the SOA, and control circuitry to control the filter to reduce a difference between a pre-determined power value of each filtered wavelength of the output signal of the SOA and the detected power value of each filtered wavelength of the output signal of the SOA.