probml / sts-jax
Structural Time Series in JAX
☆180Updated 4 months ago
Related projects: ⓘ
- Tutorials and sampling algorithm comparisons☆67Updated this week
- State of the art inference for your bayesian models.☆135Updated 3 weeks ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆220Updated 8 months ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆212Updated this week
- Gaussian processes in JAX.☆436Updated last week
- Tools for an Aesara-based PPL.☆64Updated 3 weeks ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆325Updated 5 months ago
- Recursive Bayesian Estimation (Sequential / Online Inference)☆56Updated 5 months ago
- State Space Models library in JAX☆662Updated this week
- Deep GPs built on top of TensorFlow/Keras and GPflow☆120Updated 2 months ago
- Uncertainty quantification with PyTorch☆311Updated last month
- Gaussian process modelling in Python☆215Updated 7 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆94Updated 10 months ago
- Turn SymPy expressions into trainable JAX expressions.☆313Updated 4 months ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆94Updated last year
- The tiniest of Gaussian Process libraries☆282Updated last week
- Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/☆299Updated 3 weeks ago
- Mathematical operations for JAX pytrees☆181Updated 4 months ago
- Bayesian Bandits☆64Updated last year
- This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a…☆300Updated 6 months ago
- esig python package☆41Updated 2 weeks ago
- Neat Bayesian machine learning examples☆52Updated last week
- Loopy belief propagation for factor graphs on discrete variables in JAX☆124Updated 3 months ago
- Python wrapper for nuts-rs☆106Updated this week
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆88Updated last week
- An introduction to functional programming for scalable statistical computing☆69Updated 2 months ago
- Large-scale, multi-GPU capable, kernel solver☆179Updated 2 months ago
- ☆529Updated 2 months ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆797Updated this week
- A tool-box for prior elicitation.☆78Updated this week