Thank you Alex for your reply.
I haven’t come across that link yet, so that is very helpful and seems to be what I was looking for. Yes, I suppose ideally my plan is to then tie them in with TorchScript but the link you posted details that process to some extent too.