Is there a simple way to convert a JVP into a VJP? #18635
Answered
by
patrick-kidger
NeilGirdhar
asked this question in
General
-
|
Flax has a lifted VJP, but no lifted JVP. I wonder if there's an easy way to convert one to the other? |
Beta Was this translation helpful? Give feedback.
Answered by
patrick-kidger
Nov 22, 2023
Replies: 1 comment 1 reply
-
|
|
Beta Was this translation helpful? Give feedback.
1 reply
-
|
Ah yes, thanks! I knew I had read somewhere that it was possible. Unfortunately, Flax doens't implement |
Beta Was this translation helpful? Give feedback.
Answer selected by
NeilGirdhar
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
jax.linear_transpose?