Variational Koopman models