probml / pgm-jaxLinks
Probabilistic Graphical Models in JAX
☆15Updated 2 years ago
Alternatives and similar repositories for pgm-jax
Users that are interested in pgm-jax are comparing it to the libraries listed below
Sorting:
- Jax SSM Library☆48Updated 2 years ago
- Utilities for probabilistic ML☆36Updated last year
- First-Order Probabilistic Programming Language☆28Updated 6 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 7 months ago
- Composable kernels for scikit-learn implemented in JAX.☆43Updated 4 years ago
- Bayesian Bandits☆68Updated last year
- ☆30Updated 3 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆99Updated last year
- Automatic Reparameterisation of Probabilistic Programs