Improved inference of the latent boxes, either through better optimization or through Bayesian approaches is another natural extension.
it is not generally tractable to integrate high-dimensional cones under arbitrary non-product measures.