Hi! Is anyone else having issues with this sampler when running on jupyter notebook? I spent some time debugging in spyder, then came back to the notebooks and when running import pymc as pm import pymc.sampling_jax which usually gave me an ‘experimental’ warning, I now get that the module is no…

Can you import jax in that notebook?

Sampling_jax issues

jaharvey8 July 9, 2023, 4:01pm 4

Was unaware of the nuts_sampler="numpyro" syntax, much appreciated

1 Like

Topic		Replies	Views
Jax sampling in Pymc4: no pm.sampling.jax module v5	5	1667	April 10, 2024
Confusion between latest release version and development version of PyMC Questions	7	544	October 31, 2021
Installation guide for pymc3jax v3	6	1614	March 18, 2022
Unable to get pymc3-JAX to run -- ImportError Development theano	3	1572	November 24, 2020
Omnistaging issue with JAX sampling Questions	2	1309	April 20, 2021