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

fix typo

Browse Source
This commit is contained in:
Ivy Zheng 2023-03-05 13:14:57 -08:00 committed by GitHub
parent 1a4527ed40
commit 46838a4116
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
CHANGELOG.md
View File

@ -9,7 +9,7 @@ Remember to align the itemized text with the first line of an item within a list
## jax 0.4.6
* Changes
* `jax.tree_utils` now contain a set of APIs that allow user to define keys for their
* `jax.tree_util` now contain a set of APIs that allow user to define keys for their
custom pytree node. This includes:
* `tree_flatten_with_path` that flattens a tree and return not only each leaf but
also their key paths.
@ -21,7 +21,7 @@ Remember to align the itemized text with the first line of an item within a list
* Deprecations
* The old key-path APIs in `jax.tree_util` are deprecated and will be removed 3 months
from Mar 10 2023:
* `register_keypaths`: use {func}`jax.lax.register_pytree_with_keys` instead.
* `register_keypaths`: use {func}`jax.tree_util.register_pytree_with_keys` instead.
* `AttributeKeyPathEntry` : use `GetAttrKey` instead.
* `GetitemKeyPathEntry` : use `SequenceKey` or `DictKey` instead.