New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

No module named 'jax.linear_util' #17

Open

paride92 opened this issue Jun 5, 2024 · 0 comments

paride92 commented Jun 5, 2024

Dear all,

I am trying to run your script in google colab https://colab.research.google.com/github/pyro-ppl/numpyro/blob/64fcc1ca64dae20b2f5624fb563fc18ab927b9ae/notebooks/source/tbip.ipynb

However, when I reach the "Perform Inference" section, I receive this error: ModuleNotFoundError: No module named 'jax.linear_util'

Do you have any suggestions on how to fix the issue? I tried to change the version of numpyro and jax, but I could not solve the issue.

Best regards

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment