Gain Scheduling Control of Nonlinear Systems Based on Neural State Space Models