@jspisak I filed a GitHub issue here: https://github.com/pytorch/pytorch/issues/45273 42
And linked to this PyTorch thread. Presumably we’ll continue to discuss here to determine if I’m doing something silly or if there might be a bug/missing feature (the former much more likely). Thanks!