Hi @ailzhang,
Thanks for responding !
We used the tracing API in 0.3 in the following way,

Tracing the forward pass,

trace, fwd_outputs = torch.jit.trace(model, fwd_args, nderivs = 1)


Tracing the backward pass,

torch.autograd.backward(fwd_outputs, bwd_args)

After Step 2, the trace.graph() is updated with nodes from the backward pass.
Could you please suggest an approach to generate equivalent results in the v1.1.0 API ?
Thanks,
Sanjay