Could you explain your use case a bit more?
I.e. what does serving mean in the context and what would you like to add?
Are you referring to some deployment software / platform or just a plain PyTorch model, you would like to change?