avitase:

using the nightly build I get the same error


Thank you for checking!
So if you can get the example down to where the model is a short sequential model of standard PyTorch or somesuch, either of us could do a PR to fix it.