Thanks for your reply!
Adding var to the export is a good idea . This is the first time I’ve seen it in a model I am exporting as well, but it seems like an operator that could be useful to others - I will open a feature request in PyTorch.