Model compression optimized neural network controller for nonlinear systems