One difficulty when training a VAE is the vanishing of the KL term where q ϕ (z|x) essentially becomes equal to p(z) reducing q ϕ (z|x) to a point estimate. This is known as posterior collapse and when it occurs the latent space becomes uninformative and therefore not amenable for molecular optimization.