Skip to content

Conversation

@davidwalter2
Copy link
Collaborator

@davidwalter2 davidwalter2 commented Jan 12, 2026

The only caveat is that this implementation does not work with forward differentiation i.e. one has to use --globalImpacts with --globalImpactsDisableJVP.

  • Maybe this could be fixed through implementation of custom gradients, I've tried but was not successful.

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

Successfully merging this pull request may close these issues.

1 participant