-
Notifications
You must be signed in to change notification settings - Fork 2
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
Feature/probabilistic kh example #953
Conversation
The validator prevents the class being instantiated inside a JIT loop. jaxtyping.Scalar is also added for this purpose.
The coverage check is failing (with a 0.01% reduction), which I believe is false positive. The only tested file that is affected by this PR is Should be fine to merge in my opinion. |
…reax into feature/probabilistic_kh_example
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Looking good! A few more changes, plus one comment left unresolved from the last review.
Co-authored-by: rg936672 <[email protected]>
Co-authored-by: rg936672 <[email protected]>
Ignoring coverage pipeline failure - 0.01% reduction is essentially a false positive here. |
PR Type
Description
Added an example that runs probabilistic Kernel Herding and plots the results.
How Has This Been Tested?
Existing tests pass as expected.
Does this PR introduce a breaking change?
Checklist before requesting a review