Support trained model serialization to ONNX format

[omar-araboghli]

It would be great if the conversion to ONNX IR is supported. This can most probably be done by using the scripting concept of torch.onnx.export introduced by TorchScript.

[jiesutd]

Thanks for this suggestion. We are working on a new version that will supports various model formats.