Memory allocation limit for NUTS with custom `logp` function (but not with VI methods)

Hm, it’s a surreal experience to search for examples using Print only to find my own post from a couple years ago that uses it successfully in PyMC3:

However, printing is no longer a submodule of the tensor module, and using the same syntax I did in the post I linked here doesn’t work. This to me seems like an aesara bug with Print.

As an aside, apparently you were the one that solved my problem in that previous post! :sweat_smile:

2 Likes