* WIP: Support torchscript. * Minor fixes. * Fix style issues. * Add documentation about how to deploy a trained model.