Open MikeMpapa opened 1 month ago
I recently hit this error too on a separate problem.
I think that jax just removed maps
from experimental recently. That has been deprecated for a while https://github.com/google/jax/blob/5e418f5ab2692d4791816e85ed82eb0834a579cb/CHANGELOG.md?plain=1#L284
Problem:
This problem is from the Haliax package: see here, thread_resource
has moved into jax.experimental.mesh_utils.py
with recent refactoring. They need to change that.
Solution: You can fork Haliax repo yourself, fix the importation problem and replace the Haliax link in the said docker image here with link to your yours.
Hi I am trying to use the levanter image but I get the following error: ModuleNotFoundError: No module named 'jax.experimental.maps'.
Was the model renamed? It worked fine yesterday
Thanks!
The complete error log: