flax
[nnx] Fix variable unflatten
#3578
Merged

[nnx] Fix variable unflatten #3578

cgarciae
cgarciae nnx: fix variable unflatten
0bf48f65
codecov-commenter
superbobry
superbobry commented on 2023-12-21
superbobry
superbobry approved these changes on 2023-12-21
chiamp
chiamp approved these changes on 2023-12-21
chiamp chiamp added pull ready
copybara-service copybara-service merged 2d15e85f into main 2 years ago
copybara-service copybara-service deleted the nnx-fix-variable-unflatten branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone