Skip to content

Commit

Permalink
changelog: batching rule change for rng_bit_generator
Browse files Browse the repository at this point in the history
  • Loading branch information
froystig committed Apr 4, 2024
1 parent b322d39 commit 2a36d75
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions CHANGELOG.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,11 @@ Remember to align the itemized text with the first line of an item within a list
* Changes
* Complex-valued {func}`jax.numpy.geomspace` now chooses the logarithmic spiral
branch consistent with that of NumPy 2.0.
* The behavior of `lax.rng_bit_generator`, and in turn the `'rbg'`
and `'unsafe_rbg'` PRNG implementations, under `jax.vmap` [has
changed](https://github.com/google/jax/issues/19085) so that
mapping over keys results in random generation only from the first
key in the batch.

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

0 comments on commit 2a36d75

Please sign in to comment.