PR #1629 Implement vjp and jvp transforms

Implement vjp and jvp transforms #1629

copybara-service merged 3 commits into google:main from jheek:lifted-jacobian-products

jheek requested a review from

levskaya 4 years ago

google-cla added cla: yes

Implement vjp and jvp transforms

6906619f

jheek force pushed to 6906619f 4 years ago

levskaya approved these changes on 2021-10-25

Fix pytype issue

31d04f8b

Fix docs

0a1c8905

jheek added pull ready

copybara-service merged 03ea1a74 into main 4 years ago

Reviewers

levskaya

Assignees

No one assigned

Labels

cla: yes pull ready

Milestone

No milestone