Replies: 2 comments 2 replies
-
I don't know of any easy/general way to do this. There is some relevant discussion here: #3801 |
Beta Was this translation helpful? Give feedback.
2 replies
-
Upon some investigation, I found that |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Hi all,
I recently found that JAX can efficiently calculate the Hessian-vector product (hvp) of a scalar function
f(x)
and a given vectorz
. Suppose now my function takes a variable number of arguments:f(*x)
. How to calculate the hvp off(*x)
with respect to thei
-th argumentx_i
and the corresponding given vectorz_i
?Beta Was this translation helpful? Give feedback.
All reactions