Oh, so this might be the wrong forum? I could start a thread on the PyTorch Github, if you think that’d be more appropriate!