Errors with module jax on Debian 13

Hello.

I installed pybamm via pip on a Debian 13 machine, Linux kernel 6.12.48 and with Python 3.13.5. When importing pybamm I get the error “ModuleNotFoundError: No module named ‘jax’”. After forcing the installation of jax and jaxlib with pip, now I get the error “ModuleNotFoundError: No module named ‘jax.config’”. Googling for possible solutions I found this link, apparently jax.config is deprecated. I changed that in the code and now I get “ModuleNotFoundError: No module named ‘jax.util’”. Any ideas? I am still trying to find a solution.

Thanks!

Right. I just found #5203. I guess I need to wait until these changes are uploaded to PyPi. For the moment I can try running pybamm on a docker container with Python 3.10 and see how it goes. Thanks!