- [ ] Rewrite https://github.com/rshin/seq2struct/blob/master/seq2struct/models/lstm.py to use TorchScript instead - [ ] Test all major models