value_and_grad for vector-valued functions #16959
Answered
by
jakevdp
Devan-Kerman
asked this question in
General
-
Is there such an equivalent, or is it not applicable to this scenario due to the intricacies of auto differentiation? |
Beta Was this translation helpful? Give feedback.
Answered by
jakevdp
Aug 4, 2023
Replies: 1 comment
-
Hi - thanks for the question! See #762 (comment) for some discussion of why this isn't included in JAX. The thread below that comment has some suggestions of how you might compute this in cases where you might need it. |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
Devan-Kerman
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Hi - thanks for the question!
See #762 (comment) for some discussion of why this isn't included in JAX. The thread below that comment has some suggestions of how you might compute this in cases where you might need it.