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

Open

Commits

Rename NNX view functions to have common naming convention.

samanklesaria committed 48 days ago
Add flag for variables in nnx.merge

samanklesaria committed 48 days ago
Make all NNX view functions use recursive_map or nnx.map (to handle graphs)

samanklesaria committed 48 days ago
Add deprecations for old NNX view functions

samanklesaria committed 48 days ago
Fix _to_nested_state bug

samanklesaria committed 48 days ago