Does torch.jit.trace support torch.augrad.Function

jit
1

I have a custom op using torch.autograd.Function, in which the forward and backward functions use a custom c++ op, I would like to know how to export this custom torch.augrad.Function when using torch.jit.trace