As stated in the title, pytensor.scan causes an error when sampling with sample_blackjax_nuts. The code is as follows: import pymc as pm import pytensor import pytensor.tensor as pt from pymc import sampling_jax def theano_fn(x, y): # x is the t-th value of a # y is the output of previous…

The last version of PyMC and PyTensor should work with scans and JAX already. Let us know if it doesn’t.

Error when using `sample_blackjax_nuts` with `pytensor.scan`

mathDR February 2, 2023, 7:55pm 5

@juststarted did you ever get this to work?

Topic		Replies	Views
Using theano scan with pymc3 Questions	3	887	December 4, 2020
How to use pytensor's scan function version agnostic	5	1607	February 27, 2023
Sample within theano.scan? Questions theano	5	1685	August 20, 2018
Theano.scan and pm.Model() are strangely incompatible Questions	9	1843	May 21, 2019
NameError: unbound axis name raised during transformation of variables after sample_numpyro_nuts v5 bug , jax	6	997	June 13, 2023