Identifying latent state transition in non-linear dynamical systems