Skip to content
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

Enzyme wrappers for AD utilities in Lux #738

Open
7 tasks
avik-pal opened this issue Jun 27, 2024 · 0 comments
Open
7 tasks

Enzyme wrappers for AD utilities in Lux #738

avik-pal opened this issue Jun 27, 2024 · 0 comments
Labels
enzyme good first issue Good for newcomers

Comments

@avik-pal
Copy link
Member

Functions

  • batched_jacobian
    • Enzyme.Forward
    • Enzyme.Reverse
  • vector_jacobian_product
  • jacobian_vector_product

Nested Autodiff

Enzyme needs us to use autodiff_thunk for the internal call, so we can easily replace the call inside custom rrules.

  • EnzymeRules
  • CRC.rrule
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enzyme good first issue Good for newcomers
Projects
None yet
Development

No branches or pull requests

1 participant