flax
Consolidate "view" functions with similar naming convention.
#5376
Open

Consolidate "view" functions with similar naming convention. #5376

samanklesaria wants to merge 5 commits into google:main from samanklesaria:view_renames
samanklesaria
review-notebook-app
vfdev-5
samanklesaria
samanklesaria Rename NNX view functions to have common naming convention.
d7474269
samanklesaria Add flag for variables in nnx.merge
96e05c5e
samanklesaria Make all NNX view functions use recursive_map or nnx.map (to handle g…
fbf5dbe1
samanklesaria samanklesaria force pushed from e2f9442e to fbf5dbe1 45 days ago
samanklesaria samanklesaria marked this pull request as ready for review 45 days ago
samanklesaria Add deprecations for old NNX view functions
1e6668e3
samanklesaria samanklesaria force pushed from 19266652 to 1e6668e3 45 days ago
samanklesaria Fix _to_nested_state bug
c5f45733

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone