It looks like the second case is setting all the residuals equal to zero but still contributing their logps. The masking should always be done on the probabilities (or log probs) as opposed to the actual random value variables.
It looks like the second case is setting all the residuals equal to zero but still contributing their logps. The masking should always be done on the probabilities (or log probs) as opposed to the actual random value variables.