gpjax
https://github.com/jaxgaussianprocesses/gpjax
Python
Gaussian processes in JAX.
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported1 Subscribers
Add a CodeTriage badge to gpjax
Help out
- Issues
- bug: Prediction fails for models with frozen parameters
- `DualVariationalGaussian` can leave the PD cone / NaN under plain `fit()` despite `dual_elbo` docstring calling it valid
- Split natural-gradients maths into a new Getting-started notebook
- V1.0
- Spec: didactic marimo notebook recreating arXiv:2608.04664 (soft-noncrossing Bayesian panel QR) with GPJax v1.0 + NumPyro
- docs: serve the documentation from gpjax.quantclimate.com
- `GaussianDistribution.kl_divergence` raises for `BlockDiag` and `Kronecker` covariances
- `NaturalVariationalGaussian.prior_kl` discards the triangular root it already computes
- dev: Adopt the abstract/final pattern from Equinox
- Cache the conjugate predictive representation (Cholesky + α) across `predict` calls
- Docs
- Python not yet supported