I\'m using Pyro for MCMC inference of parameters of some complicated calculation. Is there a way to keep the values of calculations made with the inferred variables?