We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
rng_bit_generator
1 parent b322d39 commit 2a36d75Copy full SHA for 2a36d75
CHANGELOG.md
@@ -25,6 +25,11 @@ Remember to align the itemized text with the first line of an item within a list
25
* Changes
26
* Complex-valued {func}`jax.numpy.geomspace` now chooses the logarithmic spiral
27
branch consistent with that of NumPy 2.0.
28
+ * The behavior of `lax.rng_bit_generator`, and in turn the `'rbg'`
29
+ and `'unsafe_rbg'` PRNG implementations, under `jax.vmap` [has
30
+ changed](https://github.com/google/jax/issues/19085) so that
31
+ mapping over keys results in random generation only from the first
32
+ key in the batch.
33
34
* Deprecations & Removals
35
* {func}`jax.tree_map` is deprecated; use `jax.tree.map` instead, or for backward
0 commit comments