Mitigating Fiber Nonlinearity with Machine Learning