Hi @dreamqin68, you can have a look at this discussion on Github:
In general, the idea is to wrap it as a theano operation and call it within with pm.Model():
Hi @dreamqin68, you can have a look at this discussion on Github:
In general, the idea is to wrap it as a theano operation and call it within with pm.Model():