Neural network based stochastic optimal control for nonlinear Markov jump systems