Hi Archie!
Here is a post with some examples that illustrate when inplace operations will and won’t break autograd:
Best.
K. Frank