Pytorch jit script error

Hi @shy,

This looks like an issue with some precondition in ONNX export. Do you mind filing an issue on github so we can route this to the right developers?

Thanks,
James