Why does io_callback in a custom JVP rule throw an error about unsupported JVP? #23488
Unanswered
mariogeiger
asked this question in
Q&A
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
I expected that by defining a custom jvp rule for my function, the io_callback used inside the custom rule wouldn’t need to support JVP itself. However, I’m getting the error: ValueError: IO callbacks do not support JVP. Could you explain why this happens and suggest how I might modify my code to call io_callback in a custom jvp?
Beta Was this translation helpful? Give feedback.
All reactions