Loopy belief propagation for factor graphs on discrete variables, in JAX!
☆65Oct 7, 2024Updated last year
Alternatives and similar repositories for PGMax
Users that are interested in PGMax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Loopy belief propagation for factor graphs on discrete variables in JAX☆164Oct 18, 2024Updated last year
- Probabilistic inference for models of behaviour☆11Mar 5, 2026Updated 2 weeks ago
- PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021☆13Oct 26, 2021Updated 4 years ago
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆65Mar 20, 2024Updated 2 years ago
- Very deep VAEs in JAX/Flax☆46Jun 16, 2021Updated 4 years ago
- Generic API for dispatch to Pyro backends.☆16Feb 13, 2022Updated 4 years ago
- A library that allows for inference on probabilistic models☆273Oct 18, 2023Updated 2 years ago
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆33Sep 15, 2023Updated 2 years ago
- Documentation:☆130May 22, 2023Updated 2 years ago
- The uncompromising Jupyter notebook formatter.☆14Jan 26, 2024Updated 2 years ago
- Sparse nonlinear least squares in JAX☆273Feb 26, 2026Updated 3 weeks ago
- ☆14Oct 7, 2022Updated 3 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated 2 years ago
- Blang's software development kit☆25Nov 26, 2025Updated 3 months ago
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆105Oct 23, 2023Updated 2 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆63Jun 17, 2024Updated last year
- Probabilistic programming system for fast and exact symbolic inference☆86Jul 10, 2024Updated last year
- A Python package for probabilistic state space modeling with JAX☆941Jan 6, 2026Updated 2 months ago
- Automatic differentiation for NumPy☆42Jun 14, 2013Updated 12 years ago
- Inference in Bayesian Networks with R☆12Feb 6, 2023Updated 3 years ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Oct 9, 2022Updated 3 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆304Updated this week
- Reference implementation of variational sequential Monte Carlo proposed by Naesseth et al. "Variational Sequential Monte Carlo" (2018)☆66Apr 29, 2019Updated 6 years ago
- A Scalable Approximate Method for Probabilistic Neurosymbolic Inference☆23Jan 27, 2025Updated last year
- Implementation of Bayesian Sum-Product Networks☆13May 19, 2020Updated 5 years ago
- Domain and problem PDDL parser in Python3 using ply.☆13Aug 21, 2018Updated 7 years ago
- Calculate indel trajectory likelihoods using methods of Miklós, Lunter & Holmes (2004), De Maio (2020), and Holmes (2020)☆14Sep 17, 2022Updated 3 years ago
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated last year
- ☆627Jan 23, 2026Updated 2 months ago
- Agent models implemented with Pyro☆11Jul 11, 2023Updated 2 years ago
- Sparse Signaling Pathway Sampling: MCMC for signaling pathway inference☆14Oct 24, 2022Updated 3 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- ☆14Jul 23, 2015Updated 10 years ago
- Julia implementation of elliptical slice sampling.☆16Dec 4, 2025Updated 3 months ago
- Fast and Differentiable Message Passing on Pairwise Markov Random Fields, ACCV 2020☆10Nov 11, 2025Updated 4 months ago
- ☆11Jun 17, 2016Updated 9 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,033Feb 3, 2026Updated last month
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated 10 months ago