Tensorflow Extended: Is it possible to use pytorch training loop in Tensorflow extended flow

Question

I have trained an image classification model using pytorch.
Now, I want to move it from research to production pipeline.
I am thinking of using TensorFlow extended.