Augmented Neural Lyapunov Control