Add VAE to test-suits

[whyhardt]

Most test-suits incorporate GAN and AE-GAN. Would be nice to cover VAE as well.

[chadcwilliams]

Agreed! I can add that in. I wasn't sure if we were going to keep VAE for the actual pip release - something to discuss in our next meet.