A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algorithms, both in a sequential and parallel fashion, as well as efficient gradient rules for computing gradients of required quantities (such as the pseudo-loglikelihood of the system).
☆14Apr 20, 2026Updated 6 months ago
Alternatives and similar repositories for sqrt-parallel-smoothers
Users that are interested in sqrt-parallel-smoothers are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Companion code in JAX for the paper Parallel Iterated Extended and Sigma-Point Kalman Smoothers.☆28Aug 9, 2024Updated 2 years ago
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated 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…☆22Jul 20, 2024Updated 2 years ago
- State-space model inference with JAX☆76Oct 2, 2026Updated last month
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆34Sep 15, 2023Updated 3 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 9 months ago
- Code accompanying the NeurIPS 2021 Paper: A Probabilistic State Space Model for Joint Inference from Differential Equations and Data (Sch…☆13Nov 7, 2022Updated 3 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- GPFlow based implementation of temporal parallellization of state space GPs☆16Sep 8, 2022Updated 4 years ago
- Computation-Aware Gaussian Processes for JAX☆13Jul 1, 2026Updated 4 months ago
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated 2 years ago
- A zoo of implementations of differential equation problems in NumPy and JAX. Oscillators, chemical reactions, n-body problems, epidemiolo…☆19Dec 28, 2023Updated 2 years ago
- Zheng Zhao's doctoral dissertation from Aalto University☆35Oct 10, 2022Updated 4 years ago
- Physics-Enhanced Regression for Initial Value Problems☆21Feb 7, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Iterative Linearized Control Toolbox☆37Aug 7, 2024Updated 2 years ago
- Source code for my PhD thesis: Backpropagation Beyond the Gradient☆21Feb 25, 2023Updated 3 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆103Jul 6, 2023Updated 3 years ago
- Various estimators of the infinite dimensional exponential family model☆16May 17, 2017Updated 9 years ago
- Gaussian processes with spherical harmonic features in JAX☆16Aug 24, 2025Updated last year
- Codes for Hilbert space reduced-rank GP regression☆15Jul 30, 2019Updated 7 years ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Neural Optimal Feedback Control☆12Mar 5, 2022Updated 4 years ago
- ☆13Feb 3, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆245Dec 22, 2023Updated 2 years ago
- Input Inference for Control (i2c), a control-as-inference framework for optimal control☆26Sep 6, 2023Updated 3 years ago
- LaTeX style file for Transactions on Machine Learning Research☆17Jun 30, 2023Updated 3 years ago
- Taylor mode automatic differentiation (jets) in PyTorch☆15Updated this week
- ☆10Jan 12, 2024Updated 2 years ago
- Repository for the paper "Bayesian Quadrature on Riemannian Data Manifolds"☆10Jun 18, 2024Updated 2 years ago
- Exponential families for JAX☆77Oct 23, 2026Updated 2 weeks ago
- An Optax-based JAX implementation of the IVON optimizer for large-scale VI training of NNs (ICML'24 spotlight)☆10Dec 19, 2024Updated last year
- ☆11Sep 3, 2022Updated 4 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Implementation of Receding Horizon Curiosity Algrithm☆13Mar 24, 2023Updated 3 years ago
- Stencil computations in JAX☆73Nov 15, 2025Updated 11 months ago
- Differential equation solvers with global error estimation☆11Aug 14, 2026Updated 2 months ago
- ☆10Jun 13, 2022Updated 4 years ago
- Implementation of point-based value iteration (for POMDPs)☆12Mar 31, 2020Updated 6 years ago
- ☆12Jul 26, 2016Updated 10 years ago
- MUBI Addon for Kodi / XBMC☆13Jan 2, 2021Updated 5 years ago