We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
jax.random.key
PRNGKey
1 parent 2a36d75 commit f247822Copy full SHA for f247822
CHANGELOG.md
@@ -30,6 +30,8 @@ Remember to align the itemized text with the first line of an item within a list
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`.
35
36
* Deprecations & Removals
37
* {func}`jax.tree_map` is deprecated; use `jax.tree.map` instead, or for backward
0 commit comments