You need to pass a theano function instead of a numpy function. Wrapping the numpy function into theano using theano.as_op might work as well but you won’t have gradient and cannot sample with NUTS.
You need to pass a theano function instead of a numpy function. Wrapping the numpy function into theano using theano.as_op might work as well but you won’t have gradient and cannot sample with NUTS.