Using torch.nn.DataParallel with torch.distributions.Laplace throws TypeError: 'Laplace' object is not iterable

Question

I am trying to train my model on multiple GPUs but I have some trouble with torch.distributions.Laplace that I call in the forward pass. I have uploaded a minim