An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.
☆22Mar 31, 2021Updated 4 years ago
Alternatives and similar repositories for SparseGPAX
Users that are interested in SparseGPAX are comparing it to the libraries listed below
Sorting:
- A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.☆33Aug 27, 2020Updated 5 years ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Dec 20, 2023Updated 2 years ago
- ☆15Jan 10, 2021Updated 5 years ago
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Sep 26, 2023Updated 2 years ago
- ☆11May 27, 2022Updated 3 years ago
- Code that accompanies the paper Guided Deep Kernel Learning☆10Jul 2, 2023Updated 2 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- Kernels, the machine learning ones☆14Mar 6, 2023Updated 2 years ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Oct 12, 2020Updated 5 years ago
- Quantifying and reporting uncertainty in drug discovery predictions with probabilistic models☆11Apr 29, 2022Updated 3 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆74Nov 13, 2020Updated 5 years ago
- [AAAI24] LE-PDE-UQ endows deep learning-based surrogate models with robust and efficient uncertainty quantification capabilities for both…☆16Feb 26, 2024Updated 2 years ago
- A lightweight didactic library of kernel methods using the back-end JAX.☆12Mar 8, 2023Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jax☆11Oct 6, 2020Updated 5 years ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆241Dec 22, 2023Updated 2 years ago
- Non-stationary spectral mixture kernels implemented in GPflow☆29Nov 30, 2018Updated 7 years ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆40Nov 19, 2022Updated 3 years ago
- Reference implementation for "Group Equivariant Subsampling"☆16Dec 13, 2021Updated 4 years ago
- Synthetic data library used in operator learning for PDE problems that overcomes dependence on classical solvers such as finite differenc…☆17Aug 8, 2024Updated last year
- Code associated with our paper "Learning Group Structure and Disentangled Representations of Dynamical Environments"☆15Dec 8, 2022Updated 3 years ago
- Probabilistic Numerics in Python.☆459Jul 3, 2025Updated 8 months ago
- A framework for composing Neural Processes in Julia☆76May 3, 2021Updated 4 years ago
- AutoML Two-Sample Test☆19Aug 3, 2022Updated 3 years ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- Measuring if attention is explanation with ROAR☆22Mar 3, 2023Updated 3 years ago
- [NeurIPS 2020] Task-Agnostic Amortized Inference of Gaussian Process Hyperparameters (AHGP)☆23Dec 9, 2020Updated 5 years ago
- Implementation for Non-stationary Spectral Kernels (NIPS 2017)☆20Jan 4, 2020Updated 6 years ago
- JAX exponential map normalising flows on sphere☆17Oct 4, 2020Updated 5 years ago
- Phylogenetic inference using Stan☆23Mar 6, 2025Updated 11 months ago
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆26Oct 28, 2021Updated 4 years ago
- Code for "Continuous-Time Meta-Learning with Forward Mode Differentiation" (ICLR 2022)☆25Mar 4, 2022Updated 3 years ago
- Implementation of the models and datasets used in "An Information-theoretic Approach to Distribution Shifts"☆25Nov 2, 2021Updated 4 years ago
- Code accompanying our paper "Feature Learning in Infinite-Width Neural Networks" (https://arxiv.org/abs/2011.14522)☆62May 11, 2021Updated 4 years ago
- Gaussian processes in JAX and Flax.☆593Feb 20, 2026Updated last week
- Robust initialisation of inducing points in sparse variational GP regression models.☆34Oct 7, 2022Updated 3 years ago
- ☆30Feb 11, 2022Updated 4 years ago
- Geometric kernels on manifolds, meshes and graphs☆291Feb 22, 2026Updated last week
- Bayesian Bandits☆69Sep 3, 2023Updated 2 years ago
- ☆17Feb 21, 2026Updated last week