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☆165Oct 18, 2024Updated last year
- Probabilistic inference for models of behaviour☆13Mar 5, 2026Updated 2 months ago
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Jul 2, 2020Updated 5 years ago
- PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021☆13Oct 26, 2021Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Jun 16, 2021Updated 4 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆18Jan 16, 2023Updated 3 years ago
- Generic API for dispatch to Pyro backends.☆16Feb 13, 2022Updated 4 years ago
- A library that allows for inference on probabilistic models☆274Oct 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:☆131May 22, 2023Updated 2 years ago
- Sparse nonlinear least squares in JAX☆283Feb 26, 2026Updated 2 months ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆331Mar 20, 2024Updated 2 years ago
- ☆14Oct 7, 2022Updated 3 years ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 10 months 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.
- E.T. Jaynes home phone.☆45Jan 11, 2021Updated 5 years 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☆86Apr 12, 2026Updated 3 weeks ago
- A Python package for probabilistic state space modeling with JAX☆958Jan 6, 2026Updated 3 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
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆314Apr 20, 2026Updated 2 weeks ago
- A repository for a universal I/O spec for TAMP, along with scripts to convert from popular specs to our spec☆13Jun 25, 2025Updated 10 months ago
- A Scalable Approximate Method for Probabilistic Neurosymbolic Inference☆24Jan 27, 2025Updated last year
- Finding the causality in biological pathways☆14Jul 12, 2023Updated 2 years ago
- Implementation of Bayesian Sum-Product Networks☆13May 19, 2020Updated 5 years ago
- Reference implementation of variational sequential Monte Carlo proposed by Naesseth et al. "Variational Sequential Monte Carlo" (2018)☆67Apr 29, 2019Updated 7 years ago
- Automatic Reparameterisation of Probabilistic Programs☆36Jun 11, 2020Updated 5 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
- Probabilistic Programming: A Powerful New Approach to Statistical Phylogenetics☆9Jan 15, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Map single-cell transcriptomes to copy number evolutionary trees.☆14Aug 18, 2024Updated last year
- Code for the paper: https://arxiv.org/pdf/2309.06979.pdf☆21Jul 29, 2024Updated last year
- Agent models implemented with Pyro☆11Jul 11, 2023Updated 2 years ago
- ☆634Apr 18, 2026Updated 2 weeks ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- Julia implementation of elliptical slice sampling.☆16Dec 4, 2025Updated 5 months ago
- ☆14Jul 23, 2015Updated 10 years ago