- [ ] Refactor any training code as necessary - [ ] Add infrastructure to save model as TorchScript - [ ] Test that model can be re-read as TorchScript in Python/Fortran and run correctly