You would probably need to define a custom NonlinearProblem subclass, as demonstrated here:
NonlinearProblem