issues with version 0.5.1

[joamatab]

Description

The latest version 0.5.1 has issues

Importerror: .venv/lib/python3.12/site-packages/jaxlib/xla_extension.so

cannot enable executable stack as shared object requires: Invalid argument

I had no issues with 0.5.0 or older methods

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

     71   return _jaxlib_version
     74 version_str = jaxlib.version.__version__
     75 version = check_jaxlib_version(
---> 76   jax_version=jax.version.__version__,
     77   jaxlib_version=jaxlib.version.__version__,
     78   minimum_jaxlib_version=jax.version._minimum_jaxlib_version)
     80 # Before importing any C compiled modules from jaxlib, first import the CPU
     81 # feature guard module to verify that jaxlib was compiled in a way that only
     82 # uses instructions that are present on this machine.
     83 import jaxlib.cpu_feature_guard as cpu_feature_guard

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

[dfm]

See more discussion here: #26781

I believe this was fixed for the next version.

[joamatab]

the next version also has issues

[dfm]

0.5.2 was a JAX-only release, there is no associated jaxlib. You can use a nightly version as discussed in the other thread, or stick with 0.5.0 until the next version with jaxlib is out. (Should be around the middle of March.)