Fix DiagonalGaussianPrior init bug
Add tests for VAE framework
Add integration tests, make use of 'make_np_rng'
Fix bug caught by new integration tests