Is there any method to solve the nonlinear partial differential equation(e g. Gross-Pitaevskii equation) in python?