-
|
I just upgraded Jax from 0.4.x to 0.5.0. Some of my tests are failing. These tests are running experiments and asserting that they're identical to golden runs, counting on JAX being deterministic. Is it expected that results of some operations will be slightly different when doing this version upgrade? |
Beta Was this translation helpful? Give feedback.
Replies: 1 comment 2 replies
-
|
You can usually get a good sense of what changed by looking at the changelog: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-5-0-jan-17-2025 For this release, the default random number generator behavior changed, as discussed there. I expect that's what you're seeing! |
Beta Was this translation helpful? Give feedback.
-
|
Also, see https://jax.readthedocs.io/en/latest/api_compatibility.html#numerics-and-randomness for a general discussion of API compatibility guarantees when it comes to randomness. TL;DR we can't guarantee random draws will never change, but we try not to change it often, and when we do change it we note it in the changelog. |
Beta Was this translation helpful? Give feedback.
-
|
Thank you! |
Beta Was this translation helpful? Give feedback.
You can usually get a good sense of what changed by looking at the changelog: https://jax.readthedocs.io/en/latest/changelog.html#jax-0-5-0-jan-17-2025
For this release, the default random number generator behavior changed, as discussed there. I expect that's what you're seeing!