Solved by shaform in post #3 
              
              
                Hi, the trace is quite long. 
But now I kind of know what the problem is. Basically, the packed sequence could not be parallelized because it could not be divided along a batch dimension. 
The solution is to put LSTM inside another module and pack the sequence inside the forward function of that mod…