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

Register custom adjoints #487

Open
avik-pal opened this issue Jan 6, 2025 · 1 comment
Open

Register custom adjoints #487

avik-pal opened this issue Jan 6, 2025 · 1 comment

Comments

@avik-pal
Copy link
Collaborator

avik-pal commented Jan 6, 2025

Useful in general but pretty much needed for things like https://discourse.julialang.org/t/how-to-compute-tanhexp-telu-function-accurately/124464/4 where applying the direct chain rule is numerically unstable

@wsmoses
Copy link
Member

wsmoses commented Jan 6, 2025

Technically the requirement is a bit deeper where we need to enable EnzymeMLIR to support custom derivatives. Probably the right approach is for code with a custom derivative, we register and emit a new op with a definition (like a special call op), whose derivative is something we register

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants