Skip to content

Conversation

@alexfikl
Copy link
Collaborator

@alexfikl alexfikl commented Mar 1, 2024

The latest version of jax updated the location of the config object:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024

I think this should work in older versions too, since jax.config.update has been there (just the explicit jax.config.config object moved)

@inducer
Copy link
Owner

inducer commented Mar 1, 2024

Thx!

inducer added a commit to inducer/pytato that referenced this pull request Mar 1, 2024
The latest version of jax updated the location of the config object:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024

Should work

h/t @alexfikl via inducer/arraycontext#252
inducer added a commit to inducer/pytato that referenced this pull request Mar 1, 2024
The latest version of jax updated the location of the config object:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024

Should work with older versions, has been deprecated a while.

h/t @alexfikl via inducer/arraycontext#252
@alexfikl alexfikl deleted the update-jax branch March 1, 2024 15:46
inducer added a commit to inducer/pytato that referenced this pull request Mar 1, 2024
The latest version of jax updated the location of the config object:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024

Should work with older versions, has been deprecated a while.

h/t @alexfikl via inducer/arraycontext#252
nkoskelo pushed a commit to nkoskelo/pytato that referenced this pull request Mar 5, 2024
The latest version of jax updated the location of the config object:
https://jax.readthedocs.io/en/latest/changelog.html#jax-0-4-25-feb-26-2024

Should work with older versions, has been deprecated a while.

h/t @alexfikl via inducer/arraycontext#252
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants