For model.logpt we are calling model.datalogpt and model.varlogpt, each of which calls aeppl.factorized_joint_logprob. It should be possible to do a single call to aeppl with all the free and observed model variables, which would save quite a lot of repeated operations on the aeppl side. For some reason this was not working when we tried it in #5166 and #4887.
|
logp_var = self.varlogpt + self.datalogpt |