Optical fiber nonlinearity compensation