ModuleNotFoundError: No module named 'jax.example_libraries'

[MohmedSoudy]

I got the following error when importing jax_unirep package

ModuleNotFoundError: No module named 'jax.example_libraries'

My script is

from jax_unirep import fit
from jax_unirep.utils import load_params

with cuda101 and my package versions are

jax==0.2.17
jax-unirep==2.2.0
jaxlib==0.1.65+cuda101

[ElArkk]

Hi Mohmed,

Could you try the solution outlined it this solved issue: #117 and report back?