mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

changelog: batching rule change for rng_bit_generator

Browse Source
This commit is contained in:
Roy Frostig 2024-04-04 16:34:10 -07:00
parent b322d399e1
commit 2a36d75285
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
CHANGELOG.md
View File

@ -25,6 +25,11 @@ Remember to align the itemized text with the first line of an item within a list
* Changes * Changes
* Complex-valued {func}`jax.numpy.geomspace` now chooses the logarithmic spiral * Complex-valued {func}`jax.numpy.geomspace` now chooses the logarithmic spiral
branch consistent with that of NumPy 2.0. 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 * Deprecations & Removals
* {func}`jax.tree_map` is deprecated; use `jax.tree.map` instead, or for backward * {func}`jax.tree_map` is deprecated; use `jax.tree.map` instead, or for backward