The deprecated config options jax_array has been removed in jax 0.4.14 (July 27, 2023), see https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-14-july-27-2023.
Thus, jax.config.update("jax_array", True) raises AttributeError: Unrecognized config option: jax_array.
We should simply remove these as they don't do anything anyway in previous recent versions of JAX.
To prevent JAX from raising the unrecognized config option error, a temporary workaround is to declare it:
jax.config.add_option("jax_array", None, None, None, None, False)
The deprecated config options
jax_arrayhas been removed in jax 0.4.14 (July 27, 2023), see https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-14-july-27-2023.Thus,
jax.config.update("jax_array", True)raisesAttributeError: Unrecognized config option: jax_array.We should simply remove these as they don't do anything anyway in previous recent versions of JAX.
To prevent JAX from raising the unrecognized config option error, a temporary workaround is to declare it: