I am having the same question, did you find an answer?
Edit:
I coudn’t find any example or explicit statement in the documentation. Here 10, it is stated that

TorchScript is a way to create serializable and optimizable models from PyTorch code

but I think in this context optimizable refers to the just-in-time compilation.
The question could also be rephrased to: Is there an alternative to the function parameters() in torch::jit::script::Module, which enables the use of optimizers?
Edit2:
I opened in issue on github, regarding the documentation for this use case: https://github.com/pytorch/pytorch/issues/17614 88