When performing sampling with sample_numpyro_nuts on a model, and specifying the var_names argument to selectively track variables, I get a mysterious error. MWE and error below. Any idea how to fix this? Or is it a bug? MWE: import numpy as np import pymc as pm import pymc.sampling_jax # True pa…

Var_names not working with sample_numpyro_nuts

junpenglao September 12, 2022, 7:28am 5

It raises an error with pm.sample(...) as well.

Topic		Replies	Views
Variable name restrictions Questions	0	374	May 1, 2021
What does the vars argument actually do in the Metropolis sampler Questions	8	708	October 28, 2018
Jax sampling - NameError: name 'auto_403213' is not defined v5	2	390	September 6, 2022
Sampler (NUTS) proposes the same value Questions	11	745	April 27, 2018
Sample_ppc fails on traces containing a subset of variables Questions from_github	0	750	September 13, 2017