This is the kind of weird output I was getting as well. Huge powers. I don’t think I can upload data for X since it is someone else’s private dataset  X is just a float64 type numpy array. I generated mu using torch.randn. You are able to reproduce the issue without my data anyway!