Mark Goodhead

Yes I think a utility method is a good solution and would address this quite simply, I think it'd be quite small as my analogous code I use in my...

https://github.com/bambinos/bambi/pull/535 Here's an example of what I was thinking of. I appreciate it's incredibly simple (literally a one liner once you've called 'predict') but this one-liner took me a few...

Just upgraded and I got the same errors as before for both pm.sample() and the jax samplers

Ah apologies, that's a bad modification from my real model to the reproduction script - thanks! With that change (original post edited) `pm.sample()` works fine and the jax samplers give...

> We should probably patch this upstream, but in the meantime you can try to run this snippet before you sample with JAX: > > ```python > import jax >...

Yes I can confirm that code snippet fixes things - thanks very much @ricardoV94!

Example script of use: ```Python import bambi as bmb import numpy as np import pandas as pd rng = np.random.default_rng(0) size = 2_000 x = rng.normal(size=size) data = pd.DataFrame( {...