Roundtrip between Relay and TorchScript

Hi,I want to use relay to modify the compute graph of pytorch, and then tranform the modified graph back to torchscript. and using the graph to continue training. How to transform a relay graph to torchscript? Has it been implemented? Any help will be apperciated.