petergchang / sarkka-jax
☆10Updated 2 years ago
Alternatives and similar repositories for sarkka-jax:
Users that are interested in sarkka-jax are comparing it to the libraries listed below
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆12Updated last month
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆58Updated 7 months ago
- A generic interface for linear algebra backends☆71Updated 6 months ago
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆11Updated 2 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆38Updated 2 months ago
- Exponential families for JAX☆60Updated last week
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆25Updated 5 months ago
- A Julia implementation of sparse Gaussian processes via path-wise doubly stochastic variational inference.☆33Updated 4 years ago
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆40Updated last year
- ☆10Updated 2 years ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Updated 5 months ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆98Updated last year
- Deep variational Bayes filter in julia using Flux☆14Updated 4 years ago
- ☆12Updated last year
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆31Updated last year
- Conditional density estimation with neural networks☆29Updated 4 months ago
- ☆36Updated 3 years ago
- ☆13Updated last year
- A simple library to run variational inference on Stan models.☆30Updated last year
- Implementation of the Gaussian Process Autoregressive Regression Model☆62Updated 3 years ago
- GPFlow based implementation of temporal parallellization of state space GPs☆15Updated 2 years ago
- ☆129Updated 3 weeks ago
- An ultra-lightweight JAX implementation of sparse Gaussian processes via pathwise sampling.☆22Updated 3 years ago
- Practical tools for quantifying how well a sample approximates a target distribution☆27Updated 4 years ago
- ☆28Updated 2 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆68Updated 4 years ago
- Turning SymPy expressions into JAX functions☆42Updated 3 years ago
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆19Updated 9 months ago
- ☆15Updated 7 years ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆99Updated last year