flax
allow captures in nnx.grad
#5484
Open

allow captures in nnx.grad #5484

copybara-service wants to merge 1 commit into main from test_903485800
copybara-service
allow captures in nnx.grad
a0a9e9d1

Login to write a write a comment.

Login via GitHub

Reviewers
No reviews
Assignees
No one assigned
Labels
Milestone