I am running the AIR example code with the exact same parameters as provided here but unable to reproduce the result of the progression of ELBO as shown in the tutorial (Fig 3, Left). The reconstruction seems fine and the count accuracy is pretty much the same, but in my experiments the ELBO has consistently converge to around 620.0 instead of above 650.0 as shown in the figure. This is kinda odd to me. Is it possible that the figure is actually showing the likelihood instead of ELBO? Or perhaps some other reason could cause the discrepancy?