Skip to content

Commit a5b8ce1

Browse files
author
jax authors
committed
Merge pull request #20598 from froystig:changelog
PiperOrigin-RevId: 622021033
2 parents 55233a0 + f247822 commit a5b8ce1

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

CHANGELOG.md

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,13 @@ 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.
33+
* Docs now use `jax.random.key` for construction of PRNG key arrays
34+
rather than `jax.random.PRNGKey`.
2835

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

0 commit comments

Comments
 (0)