How to write a Premiere Pro/After Effects plugin using Pytorch Model

Hello,

I have trained a Pytorch model that increases the visual quality of an image by removing different visual artifacts.
I want to write a plugin for Premiere Pro/After Effect that uses this model. I have searched on Google but I can't find a complete tutorial explaining how to do it in a clean and fast way.
I only need to have a template code where for an input video, the plugin applies the model as a function on each frame and returns the sequence with the function applied.  

In addition, the audio must be correctly aligned with the output frames so that there is no offset.

Do you have any resources/documentation/tutorial/website that you can recommend? 

Thanks