Nonlinear Invariant Risk Minimization