Using Pytorch neural network as the custom “black box” likelihood function

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