Florian

pymc.sampling_jax.sample_numpyro_nuts raises AttributeError "module 'jaxlib.mlir.dialects.mhlo' has no attribute 'ConstOp'" during compilation in GaussianRandomWalk model

1

## Description of your problem When using `sampling_jax.sample_numpyro_nuts` for a timeseries model of a GaussianRandomWalk, I am getting the following error `AttributeError: module 'jaxlib.mlir.dialects.mhlo' has no attribute 'ConstOp'`. I searched...

pass names from RV dims to change_rv_size

3

@canyon289 to resolve #5669, I have made some small changes mainly to aesaraf.py to include the dim names in the new_size tensor (https://github.com/pymc-devs/pymc/commit/6585f9021a17a7f0c9c833bf4700fab240d2164b) This solves the issue mainly and `test_model`...

handle deterministic input with SNLE

2

Assume I have a model that takes deterministic input $X$ and unknown parameters $\theta$. Is it possible to train an SLN estimator on the parameters and consider the observations i.i.d...