Skip to content

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