Simon sorry to bug you but this multivariate normal thing still doesn’t work:
AttributeError: module 'torch.distributions' has no attribute 'MultivariateNormal'

note the other error:
AttributeError: module 'torch.distributions' has no attribute 'multivariate_normal'