Loopy belief propagation for factor graphs on discrete variables, in JAX!
☆66Oct 7, 2024Updated 2 years ago
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☆166Oct 18, 2024Updated 2 years ago
- Probabilistic inference for models of behaviour☆13Mar 5, 2026Updated 8 months ago
- Exponential families for JAX☆77Oct 23, 2026Updated 2 weeks ago
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 5 years ago
- PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021☆13Oct 26, 2021Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- u-MPS implementation and experimentation code used in the paper Tensor Networks for Probabilistic Sequence Modeling (https://arxiv.org/ab…☆19Jul 2, 2020Updated 6 years ago
- Source-to-Source Debuggable Derivatives in Pure Python☆15Jan 23, 2024Updated 2 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆331Mar 20, 2024Updated 2 years ago
- ☆14Jul 23, 2015Updated 11 years ago
- A library that allows for inference on probabilistic models☆275Oct 18, 2023Updated 3 years ago
- Generic API for dispatch to Pyro backends.☆16Feb 13, 2022Updated 4 years ago
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆66Mar 20, 2024Updated 2 years ago
- Downstream insertion tasks with NCF for tracking extrinsic contacts with tactile sensing☆19May 6, 2024Updated 2 years ago
- Sum-product networks in Julia.☆36Dec 13, 2022Updated 3 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.☆245Dec 22, 2023Updated 2 years ago
- Sparse nonlinear least squares in JAX☆306Jun 22, 2026Updated 4 months ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆65Jun 17, 2024Updated 2 years ago
- A Python package for probabilistic state space modeling with JAX☆1,023Updated this week
- [IROS2022] Project page for Scene-level tracking and reconstruction without object-prior.☆14Feb 3, 2024Updated 2 years ago
- The uncompromising Jupyter notebook formatter.☆14Jan 26, 2024Updated 2 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
- Bayesian model reduction for probabilistic machine learning☆12Jul 3, 2025Updated last year
- ☆15Oct 7, 2022Updated 4 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Documentation:☆132May 22, 2023Updated 3 years ago
- SE(3) exponential map at zero helper for SymPy☆34May 18, 2023Updated 3 years ago
- ☆658Oct 20, 2026Updated 2 weeks ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- Dive into Jax, Flax, XLA and C++☆32Apr 1, 2020Updated 6 years ago
- BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.☆1,138Updated this week
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆106Oct 23, 2023Updated 3 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Feb 26, 2023Updated 3 years ago
- A curated collection of papers on probabilistic circuits, computational graphs encoding tractable probability distributions.☆51Feb 6, 2024Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- This is a collection of code samples aimed at illustrating temporal parallelization methods for sequential data.☆34Sep 15, 2023Updated 3 years ago
- 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
- Blang's software development kit☆25Nov 26, 2025Updated 11 months ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆326Oct 14, 2026Updated 3 weeks ago
- Probabilistic programming system for fast and exact symbolic inference☆86Apr 12, 2026Updated 6 months ago
- ☆54Aug 22, 2023Updated 3 years ago