flax
Have _graph_flatten respect nnx.data declarations (extension of #5140)
#5159
Open
Open
Have _graph_flatten respect nnx.data declarations (extension of #5140) #5159
samanklesaria
wants to merge
2 commits into
google:main
from
samanklesaria:pr/5140
flatten respect nnx.data
6dd9baa8
Fix type error
5e977f00
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub