You can also define a gradient Op that gives you the pytorch gradients, since pytorch is also auto-diffeable. It’s shown in the blackbox examples
1 Like
You can also define a gradient Op that gives you the pytorch gradients, since pytorch is also auto-diffeable. It’s shown in the blackbox examples