File tree Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Expand file tree Collapse file tree 1 file changed +7
-0
lines changed Original file line number Diff line number Diff line change @@ -25,6 +25,13 @@ Remember to align the itemized text with the first line of an item within a list
25
25
* Changes
26
26
* Complex-valued {func}` jax.numpy.geomspace ` now chooses the logarithmic spiral
27
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
+ * Docs now use ` jax.random.key ` for construction of PRNG key arrays
34
+ rather than ` jax.random.PRNGKey ` .
28
35
29
36
* Deprecations & Removals
30
37
* {func}` jax.tree_map ` is deprecated; use ` jax.tree.map ` instead, or for backward
You can’t perform that action at this time.
0 commit comments