about doing this well this is this figure than the simple black box approach one thing we could do is model the distribution of Phi given theta as Gaussian and use the same sort of variational inference for training as I mentioned before but have her inference that work you perform an optimization over Phi and this is what was done in Ravi and pizza in 2019 another thing that we want to do is if you don't want to model p5 theta I give PFI given theta as Gaussian we can use stein variational gradient to on the last layer of the