flax
Add simple argument-only lifted nn.grad function.
#3463
Merged

Add simple argument-only lifted nn.grad function. #3463

copybara-service merged 1 commit into google:main from levskaya:vjp_fix
levskaya
levskaya levskaya requested a review from cgarciae cgarciae 2 years ago
levskaya levskaya added pull ready
levskaya levskaya force pushed 2 years ago
cgarciae
cgarciae commented on 2023-11-06
cgarciae
cgarciae commented on 2023-11-06
cgarciae
cgarciae commented on 2023-11-06
levskaya levskaya force pushed 2 years ago
levskaya levskaya force pushed 2 years ago
cgarciae
cgarciae approved these changes on 2023-11-06
levskaya levskaya force pushed 2 years ago
levskaya Add simple argument-only lifted nn.grad and nn.value_and_grad functions.
5ff36ba4
levskaya levskaya force pushed to 5ff36ba4 2 years ago
codecov-commenter
copybara-service copybara-service merged e2c3dfd3 into main 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone