AttributeError: partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import)

[mvds314]

Description

I get an error on importing jax in the Spyder IDE. In a normal python or ipython terminal it seems to work fine.
I am not 100% whether it's a Spyder issue or a jax issue.

With jax 0.5.3 everything works fine, as soon as I upgrade to 0.6 or higher I get the error below.

Traceback (most recent call last):

  Cell In[2], line 1
    import jax

  File C:\...\WPy64-31290\python\Lib\site-packages\jax\__init__.py:25
    from jax._src.cloud_tpu_init import cloud_tpu_init as _cloud_tpu_init

  File C:\...\WPy64-31290\python\Lib\site-packages\jax\_src\cloud_tpu_init.py:20
    from jax._src import config

  File C:\...\WPy64-31290\python\Lib\site-packages\jax\_src\config.py:27
    from jax._src.lib import guard_lib

  File C:\...\WPy64-31290\python\Lib\site-packages\jax\_src\lib\__init__.py:76
    jax_version=jax.version.__version__,

AttributeError: partially initialized module 'jax' has no attribute 'version' (most likely due to a circular import)

import jax

System info (python version, jaxlib version, accelerator, etc.)

Winpython 3.13.3
jax 0.6.1

[YousefElbrolosy]

It usually helps to try this in a fresh environment like conda and then reinstall your desired versions

[jakevdp]

We've seen reports like this before on Windows, . I wonder if there's some difference between the import resolver in Spyder and the import resolver in standard Python environments?

On its face, the error you're seeing doesn't make sense: jax.version is explicitly imported in this line:

jax/jax/_src/lib/__init__.py

Line 33 in 382506f

import jax.version

And then later in the file, referencing jax.version leads to this attribute error on your system, but not on any of the systems in we use for CI testing (including windows, linux, and osx). That makes me wonder what might be different about Spyder.

Maybe if we did import jax.version as jax_version and then referred to jax_version directly further down, it might resolve the issue. Is that something you can try on your system?

[jakevdp]

Another possibility is that you have some other file named jax in your import path, and removing it will fix things. See #21738 for an example where this was the case.

[mvds314]

I'm not sure what went wrong earlier. It seems to work again at the moment, maybe I changed or reinstalled things in the meantime..