Symbolically trace LSTM layers

torch.compile FX (Functional Transformations)
1

Is there any practical way to symbolically trace LSTM layers?
Should I create my custom LSTM with static sequence size?

2

Yes that sounds about right

3

Thank you for your reply @marksaroufim
And then is it exportable to ONNX ?