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`

ricardoV94 January 12, 2023, 10:25pm 2

At the moment we don’t support scan in the JAX backend. We hope to fix this soon

1 Like

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