mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 13:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

changelog: note doc change to use jax.random.key over PRNGKey

Browse Source
This commit is contained in:
Roy Frostig 2024-04-04 16:38:08 -07:00
parent 2a36d75285
commit f247822977
1 changed files with 2 additions and 0 deletions

2
CHANGELOG.md

@ -30,6 +30,8 @@ Remember to align the itemized text with the first line of an item within a list
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.
* Docs now use `jax.random.key` for construction of PRNG key arrays
rather than `jax.random.PRNGKey`.
* Deprecations & Removals
* {func}`jax.tree_map` is deprecated; use `jax.tree.map` instead, or for backward