error on jax.config import

[dgleich]

Using phiflow 2.5.3 and Phiml 1.2.1 and Jax 0.4.25 and Jaxlib 0.4.25

I got this error

ImportError: cannot import name 'config' from 'jax.config' 

which was from this line in File "./lib/python3.11/site-packages/phiml/backend/jax/_jax_backend.py", line 21

The line is

   from jax.config import config

and changing it to

   from jax import config

fixes it.

I can submit a pull request if needed.