Skip to content

fix: filter also for new (v0.7.0) LinearizeTracer type#3586

Merged
ianna merged 1 commit intomainfrom
pfackeldey/fix_jax_070
Jul 24, 2025
Merged

fix: filter also for new (v0.7.0) LinearizeTracer type#3586
ianna merged 1 commit intomainfrom
pfackeldey/fix_jax_070

Conversation

@pfackeldey
Copy link
Collaborator

Closes #3585

@pfackeldey
Copy link
Collaborator Author

Ref: https://docs.jax.dev/en/latest/changelog.html#jax-0-7-0-july-22-2025

I have to say though... the way our JAX backend works is rather fragile. I wouldn't be surprised about more issues like this in the future. (We should think about it again.)

@pfackeldey pfackeldey requested a review from ianna July 22, 2025 22:02
@ianna
Copy link
Member

ianna commented Jul 23, 2025

Ref: https://docs.jax.dev/en/latest/changelog.html#jax-0-7-0-july-22-2025

I have to say though... the way our JAX backend works is rather fragile. I wouldn't be surprised about more issues like this in the future. (We should think about it again.)

Yep, totally agree. Thanks for fixing it!

Copy link
Member

@ianna ianna left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@pfackeldey - Thanks for fixing it. We need to discuss what we should do with JAX backend.

@ianna ianna added the pr-next-release Required for the next release label Jul 24, 2025
@ianna ianna merged commit 4102a4a into main Jul 24, 2025
43 checks passed
@ianna ianna deleted the pfackeldey/fix_jax_070 branch July 24, 2025 15:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

pr-next-release Required for the next release

Projects

None yet

Development

Successfully merging this pull request may close these issues.

JAX 0.7.0 incompatibility

2 participants