flax
Have _graph_flatten respect nnx.data declarations
#5136

Open

Have _graph_flatten respect nnx.data declarations #5136

samanklesaria wants to merge 2 commits into google:main from samanklesaria:issues/5128

Have _graph_flatten respect nnx.data declarations

84319e9a

Use Modules instead of dicts in graph_utils_test

755b6825

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone