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.☆596Mar 2, 2026Updated last week
- The tiniest of Gaussian Process libraries☆333Updated this week
- Exponential families for JAX☆76Updated this week
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,025Feb 3, 2026Updated last month
- Gaussian process modelling in Python☆226Dec 15, 2024Updated last year
- Code for NeurIPS 2021 paper 'Spatio-Temporal Variational Gaussian Processes'☆48Nov 11, 2021Updated 4 years ago
- 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☆935Jan 6, 2026Updated 2 months ago
- Fully Bayesian Inference in GPs - Gaussian and Generic Likelihoods☆22Sep 26, 2023Updated 2 years ago
- Mathematical operations for JAX pytrees☆209Dec 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
- ☆624Jan 23, 2026Updated last month
- Probabilistic Numerics in Python.☆459Jul 3, 2025Updated 8 months ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆1,028Dec 17, 2025Updated 2 months ago
- Minimal Gaussian process library in JAX with a simple (custom) approach to state management.☆12Dec 20, 2023Updated 2 years ago
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆13Dec 4, 2024Updated last year
- 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☆87Nov 30, 2023Updated 2 years ago
- Tutorials and sampling algorithm comparisons☆80Updated this week
- Structural Time Series in JAX☆214May 8, 2024Updated last year
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated last year
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Feb 27, 2026Updated last week
- 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☆272Feb 26, 2026Updated last week
- Infinite-horizon Gaussian processes☆33Sep 28, 2020Updated 5 years ago
- Gradient-informed particle MCMC methods☆12Jan 29, 2024Updated 2 years ago
- Probabilistic Programming and Nested sampling in JAX☆223Feb 27, 2026Updated last week
- Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.☆2,612Feb 28, 2026Updated last week
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated last year
- ☆774Jan 27, 2024Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- State of the art inference for your bayesian models.☆233Jan 14, 2026Updated last month
- ☆233Sep 3, 2023Updated 2 years ago