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:
- Utilities for probabilistic ML☆36Updated 2 years ago
- Bayesian Bandits☆68Updated 2 years ago
- Loopy belief propagation for factor graphs on discrete variables, in JAX!☆65Updated 11 months ago
- Composable kernels for scikit-learn implemented in JAX.☆44Updated 4 years ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆100Updated 2 years ago
- Jax SSM Library☆49Updated 2 years ago
- Evaluation Framework for Probabilistic Programming Languages☆103Updated last year
- Python implementation of projection losses.☆27Updated 5 years ago
- Functional machine learning for fun☆85Updated 4 years ago
- Discontinuous Hamiltonian Monte Carlo in JAX☆42Updated 5 years ago
- Bayesian Regression Models in Pyro☆72Updated last year
- Progress, Notes, Summaries and a lot of Questions on Machine Learning☆55Updated 5 years ago
- Repository of models in Pyro☆30Updated last year
- Performant, differentiable reinforcement learning☆123Updated last month
- Dirichlet Process K-means☆48Updated last year
- ☆30Updated 3 years ago
- A package for Safe Anytime Valid Inference☆26Updated 10 months ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆33Updated 2 years ago
- Library for learning and inference with Sum-product Networks utilizing TensorFlow 2.x and Keras☆48Updated 3 weeks ago
- A toolbox for inference of mixture models☆16Updated 2 years ago
- Dependent multinomials made easy: stick-breaking with the Pólya-gamma augmentation☆61Updated 4 years ago
- A Visual Exploration of Gaussian Processes☆105Updated 2 years ago
- An interactive visualization tool that transforms probabilistic programming models into an "Interactive Probabilistic Models Explorer".☆24Updated last year
- Loopy belief propagation for factor graphs on discrete variables in JAX☆156Updated 11 months ago
- Automatic differentiation + optimization☆103Updated 6 years ago
- Pyro models and misc examples.☆19Updated 4 years ago
- ☆29Updated last year
- ☆192Updated 3 months ago
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆237Updated last year
- python app for doing personalized causal medicine using the methods invented by Judea Pearl et al.☆25Updated 2 years ago