Replies: 1 comment 3 replies
-
I don't think the gradient is well-defined for the |
Beta Was this translation helpful? Give feedback.
3 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
It is possible to write sampling functions in two ways. In one case, the key is passed into the function and the sampling happens inside the function. And in another case, the sampling happens outside the function.
Computing the grad is quite obvious in the second case. I was wondering how to write the same for the first case (or even if it is possible).
I was trying to write a
custom_jvp
:and evaluate as follows:
but I think I am doing something wrong, because it returns 0 as the gradient.
Beta Was this translation helpful? Give feedback.
All reactions