Skip to content

Commit 2a36d75

Browse files
committed
changelog: batching rule change for rng_bit_generator
1 parent b322d39 commit 2a36d75

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

CHANGELOG.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,11 @@ Remember to align the itemized text with the first line of an item within a list
2525
* Changes
2626
* Complex-valued {func}`jax.numpy.geomspace` now chooses the logarithmic spiral
2727
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.
2833

2934
* Deprecations & Removals
3035
* {func}`jax.tree_map` is deprecated; use `jax.tree.map` instead, or for backward

0 commit comments

Comments
 (0)