Hi,
When you backprop a function that is a function of the gradients, you actually ask for higher order derivatives as you need to backprop through the backward pass itself.
Unfortunately all the Functions don’t support that.
You can enable the anomaly detection 32 mode to know which forward function caused this issue.