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

Add ggn and ggnvp #75

Merged
merged 5 commits into from
Apr 24, 2024
Merged

Add ggn and ggnvp #75

merged 5 commits into from
Apr 24, 2024

Conversation

SamDuffield
Copy link
Contributor

Discussed in #63

So far I've added ggnvp for vector products. It should be fairly straightforward to add ggn too now.

Might want to consider setting normalize=False and mention that normalize=True assumes that the first dimension of the output of the jvp is batchsize (or dimension or desired normalisation).

@SamDuffield SamDuffield marked this pull request as ready for review April 23, 2024 13:56
@SamDuffield SamDuffield requested a review from KaelanDt April 23, 2024 13:56
posteriors/utils.py Show resolved Hide resolved
Copy link
Contributor

@KaelanDt KaelanDt left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Love the examples!

posteriors/utils.py Show resolved Hide resolved
@SamDuffield SamDuffield merged commit ecfeff7 into main Apr 24, 2024
2 checks passed
@SamDuffield SamDuffield deleted the ggn branch April 24, 2024 11:38
@SamDuffield SamDuffield mentioned this pull request Apr 24, 2024
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.

2 participants