I have a log-prob function and its gradient. Because FFT is involved in the log-prob function and its gradient, it is not possible to perform autodiff. So I do the differential by hand.
Is there a (documented) way to call the NUTS sampler directly?
I did notice there is a way to create a customized distribution, which however does not mention how to supply the gradient.