Similar to #1003, but for the forward and reverse mode autograd transforms.
Similar to #1003, but for the forward and reverse mode autograd transforms.