AaltoML / BayesNewtonView external linksLinks
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
Alternatives and similar repositories for BayesNewton
Users that are interested in BayesNewton are comparing it to the libraries listed below
Sorting:
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 2 years ago
- Gaussian processes in JAX and Flax.☆588Updated this week
- The tiniest of Gaussian Process libraries☆333Feb 8, 2026Updated last week
- Exponential families for JAX☆76Feb 1, 2026Updated 2 weeks ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Jul 20, 2024Updated last year
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,014Feb 3, 2026Updated last week
- Code for NeurIPS 2021 paper 'Spatio-Temporal Variational Gaussian Processes'☆47Nov 11, 2021Updated 4 years ago
- Gaussian process modelling in Python☆225Dec 15, 2024Updated last year
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Mar 31, 2021Updated 4 years ago
- A Python package for probabilistic state space modeling with JAX☆924Jan 6, 2026Updated last month
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Sep 26, 2023Updated 2 years ago
- Mathematical operations for JAX pytrees☆208Dec 5, 2024Updated last year
- Recyclable Gaussian Processes☆11Dec 8, 2022Updated 3 years ago
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆27Aug 9, 2024Updated last year
- ☆622Jan 23, 2026Updated 3 weeks ago
- Probabilistic Numerics in Python.☆458Jul 3, 2025Updated 7 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,026Dec 17, 2025Updated last month
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆12Dec 4, 2024Updated last year
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Dec 20, 2023Updated 2 years ago
- A Tensorflow based library for Time Series Modelling with Gaussian Processes☆32Jul 4, 2024Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated last year
- Normalizing Flows using JAX☆86Nov 30, 2023Updated 2 years ago
- Tutorials and sampling algorithm comparisons☆79Updated this week
- Structural Time Series in JAX☆214May 8, 2024Updated last year
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆298Jan 9, 2026Updated last month
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated last year
- Deep GPs built on top of TensorFlow/Keras and GPflow☆128Oct 16, 2024Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆74Nov 13, 2020Updated 5 years ago
- Sparse nonlinear least squares in JAX☆270Jan 21, 2026Updated 3 weeks ago
- Infinite-horizon Gaussian processes☆33Sep 28, 2020Updated 5 years ago
- Probabilistic Programming and Nested sampling in JAX☆221Dec 15, 2025Updated 2 months ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,600Updated this week
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆329Mar 20, 2024Updated last year
- ☆775Jan 27, 2024Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 2 years ago
- State of the art inference for your bayesian models.☆232Jan 14, 2026Updated last month
- ☆233Sep 3, 2023Updated 2 years ago