JaxGaussianProcesses / GPJax

Gaussian processes in JAX.
https://docs.jaxgaussianprocesses.com/
Apache License 2.0
436 stars 51 forks source link

Fix transform #464

Closed thomaspinder closed 4 weeks ago

thomaspinder commented 1 month ago

Type of changes

Checklist

Description

This PR uses replace to update the value of parameter inside transform. Through previous discussions, failure to do this means that transform becomes an impure function.

Issue Number: N/A