JIT trace doesn't work for language models

[sdrobert]

As the title says. I can't figure out how to call methods in a module which calls other methods which need to be scripted. The "self" argument will be assumed to be a tensor.

Not sure if there's a way to do this already. It's not really high-priority as you can always script the module instance.