AaltoML / kalman-jaxLinks
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
☆102Updated 2 years ago
Alternatives and similar repositories for kalman-jax
Users that are interested in kalman-jax are comparing it to the libraries listed below
Sorting:
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆239Updated last year
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Updated 2 years ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆33Updated 2 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆73Updated 4 years ago
- Documentation:☆124Updated 2 years ago
- Jax SSM Library☆48Updated 2 years ago
- Normalizing Flows using JAX☆85Updated last year
- Large-scale, multi-GPU capable, kernel solver☆191Updated 3 months ago
- Gaussian process modelling in Python☆224Updated 10 months ago
- Deep GPs built on top of TensorFlow/Keras and GPflow☆127Updated last year
- Code repo for "Kernel Interpolation for Scalable Online Gaussian Processes"☆64Updated 4 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated last year
- Exponential families for JAX☆75Updated last week
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 4 years ago
- Extensible Tensorflow library for differentiable particle filtering. ICML 2021.☆42Updated 2 years ago
- Tools for JAX☆49Updated last week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆115Updated 3 weeks ago
- Implementation of the Gaussian Process Autoregressive Regression Model☆68Updated 9 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Code for "Infinitely Deep Bayesian Neural Networks with Stochastic Differential Equations"☆170Updated 3 years ago
- ☆30Updated 3 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- A generic interface for linear algebra backends☆74Updated 7 months ago
- Implementation of stochastic variational inference for differentially deep gaussian processes☆15Updated 6 years ago
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆27Updated last year
- Recursive Bayesian Estimation (Sequential / Online Inference)☆59Updated last year
- Manifold Markov chain Monte Carlo methods in Python☆234Updated last month
- Turning SymPy expressions into JAX functions☆45Updated 4 years ago
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆42Updated 2 years ago
- Just a little MCMC☆232Updated last year