Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
☆103Jul 6, 2023Updated 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆242Dec 22, 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
- Chirp instantaneous frequency estimation using stochastic differential equation Gaussian processes☆13Oct 30, 2024Updated last year
- Prototypes of differentiable differential equation solvers in JAX.☆27Feb 3, 2020Updated 6 years ago
- Composable kernels for scikit-learn implemented in JAX.☆46Oct 26, 2020Updated 5 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Physics-Enhanced Regression for Initial Value Problems☆20Feb 7, 2024Updated 2 years ago
- Gaussian processes in JAX and Flax.☆608Updated this week
- Efficient SDE samplers including Gaussian-based probabilistic solvers. Written in JAX.☆10Feb 8, 2025Updated last year
- GPFlow based implementation of temporal parallellization of state space GPs☆16Sep 8, 2022Updated 3 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- A Tensorflow based library for Time Series Modelling with Gaussian Processes☆31Jul 4, 2024Updated last year
- Layered distributions using FLAX/JAX☆10Dec 13, 2020Updated 5 years ago
- Code accompanying VarGrad: A Low-Variance Gradient Estimator for Variational Inference☆12Oct 12, 2020Updated 5 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Feb 24, 2020Updated 6 years ago
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Gaussian process time-frequency analysis toy examples☆17Apr 5, 2019Updated 7 years ago
- Unofficial implementation of NeRF: Representing Scenes as Neural Radiance Fields for View Synthesis, using Flax with the Linen API☆13Sep 25, 2021Updated 4 years ago
- Gaussian process modelling in Python☆226Dec 15, 2024Updated last year
- Unifying probabilistic models for time-frequency analysis. Spectral Mixture Gaussian processes in state space form.☆19May 30, 2019Updated 6 years ago
- A Python package for probabilistic state space modeling with JAX☆950Jan 6, 2026Updated 3 months ago
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated last year
- Example code to speed up GP inference with gradients for high-dimensional inputs.☆14Jul 12, 2021Updated 4 years ago
- Implementation of the Gaussian Process Autoregressive Regression Model☆70Jan 17, 2025Updated last year
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Wordpress hosting with auto-scaling - Free Trial • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- End-to-End Probabilistic Inference for Nonstationary Audio Analysis☆12Aug 7, 2019Updated 6 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…☆20Jul 20, 2024Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- stan2tfp is a lightweight interface to the TensorFlow Probability backend of the Stan compiler. It provides the necessary objects and fun…☆25Jan 10, 2022Updated 4 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,051Updated this week
- Approximate Inference Turns Deep Networks into Gaussian Processes (dnn2gp)☆48Dec 17, 2019Updated 6 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆57Mar 9, 2026Updated last month
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆33Sep 15, 2023Updated 2 years ago
- Framework-agnostic library for checking array/tensor shapes at runtime.☆46May 4, 2021Updated 4 years ago
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Code used in paper submitted to ICASSP 2019☆14Nov 22, 2018Updated 7 years ago
- Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX☆23Mar 6, 2023Updated 3 years ago
- ☆631Apr 1, 2026Updated 2 weeks ago
- Documentation:☆130May 22, 2023Updated 2 years ago
- "Parameter origami" -- folding and unfolding collections of parameters for optimization and sensitivity analysis.☆14Jan 23, 2024Updated 2 years ago
- "Variational inference tools to leverage estimator sensitivity."☆15May 11, 2023Updated 2 years ago
- Exponential families for JAX☆77Updated this week