-
Notifications
You must be signed in to change notification settings - Fork 13
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Vector / chunked / batch mode #312
Comments
gdalle
added
backend
Related to one or more autodiff backends
core
Related to the core utilities of the package
labels
Jun 8, 2024
Partially solved by #320, still needs:
|
It would be nice to avoid the distinction between |
Further improved by #412, which makes batch mode the default that extensions need to implement |
Merged
Batch mode for Enzyme is nearly complete since #471, it only needs EnzymeAD/Enzyme.jl#1883 to work with out-of-place functions |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
This one is rather easy in principle: just wrap the JVP/VJP/HVP seed in a Batch struct that is essentially a named tuple, and dispatch on that
The text was updated successfully, but these errors were encountered: