Skip to content

value_and_grad for vector-valued functions #16959

Answered by jakevdp
Devan-Kerman asked this question in General
Discussion options

You must be logged in to vote

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.

Replies: 1 comment

Comment options

You must be logged in to vote
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
Labels
None yet
2 participants