lmao14

Hi, I try to reproduce the GPflow example ([https://gpflow.github.io/GPflow/2.5.2/notebooks/advanced/heteroskedastic.html](https://gpflow.github.io/GPflow/2.5.2/notebooks/advanced/heteroskedastic.html)) of heteroskedastic GP with an integration of gpytorch and pyro. Although the model fits the data well,  the model parameters...

# 🐛 Bug ## To reproduce ** Code snippet to reproduce ** ```python import torch import gpytorch import linear_operator kern = gpytorch.kernels.ScaleKernel(gpytorch.kernels.RBFKernel(batch_shape=torch.Size([4, 3]),), batch_shape=torch.Size([4, 3])) X = torch.randn([2, 5]) kxx...