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

Allow to pass trainable inducing inputs to AbstractVariationalGaussian #485

Conversation

stefanocortinovis
Copy link
Contributor

Checklist

  • [ x ] I've formatted the new code by running hatch run dev:format before committing.
  • [ NA ] I've added tests for new code.
  • [ NA ] I've added docstrings for the new code.

Description

As things stand, AbstractVariationalGaussian defaults to static inducing_inputs and making them trainable requires manual surgery. This PR enables passing trainable (i.e. Real) or fixed (i.e. Static) inducing_inputs to AbstractVariationalGaussian. When an array is passed as inducing_inputs, it is wrapped in Real by default, to mimic the default behaviour in other GP libraries.

I am happy to write unit tests if required.

Issue Number: N/A

@thomaspinder thomaspinder merged commit 7c645c2 into JaxGaussianProcesses:main Nov 1, 2024
11 checks passed
@stefanocortinovis stefanocortinovis deleted the trainable_inducing_inputs branch November 1, 2024 15:52
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