ImportError: cannot import name 'ShapedArray' from 'jax'
YR57 opened this issue · comments
When running the UnLoc code, an error occurs because the versions of jax, jaxlib, and flax do not match.
What versions of jax, jaxlib, and flax are suitable for UnLoc code?
Got the same issue. How did you solve this?