Check in variational inference and elbo code.