I’m thinking about opening an issue in Github, just to make sure PT devs have this under their radar. Hopefully the necessary change is easy to implement (although I have no idea on how TorchScript internals work…)