flax
Exposed OptState in nnx module
#4788
Merged

Exposed OptState in nnx module #4788

copybara-service merged 1 commit into main from add
vfdev-5
vfdev-5 Exposed OptState in nnx module
755067df
vfdev-5 vfdev-5 requested a review from cgarciae cgarciae 208 days ago
cgarciae
cgarciae approved these changes on 2025-06-24
cgarciae cgarciae added pull ready
copybara-service copybara-service merged 0b9bb4ee into main 208 days ago
copybara-service copybara-service deleted the add branch 208 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone