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
Sorting:
- Loopy belief propagation for factor graphs on discrete variables in JAX☆164Oct 18, 2024Updated last year
- 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
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆65Mar 20, 2024Updated last year
- PyTorch implementation for "HyperSPNs: Compact and Expressive Probabilistic Circuits", NeurIPS 2021☆13Oct 26, 2021Updated 4 years ago
- Generic API for dispatch to Pyro backends.☆16Feb 13, 2022Updated 4 years ago
- A C++ implementation (with Python bindings) of the Persistence Filter.☆18Jul 19, 2022Updated 3 years ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Documentation:☆129May 22, 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
- A library that allows for inference on probabilistic models☆273Oct 18, 2023Updated 2 years ago
- Sum-product networks in Julia.☆37Dec 13, 2022Updated 3 years ago
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆330Mar 20, 2024Updated last year
- Inference in Bayesian Networks with R☆12Feb 6, 2023Updated 3 years ago
- Very deep VAEs in JAX/Flax☆46Jun 16, 2021Updated 4 years ago
- Automatic Reparameterisation of Probabilistic Programs☆36Jun 11, 2020Updated 5 years ago
- Bayesian learning and inference for state space models (SSMs) using Google Research's JAX as a backend☆63Jun 17, 2024Updated last year
- Painless optimisation of constrained variables in AutoGrad, TensorFlow, PyTorch, and JAX☆23Mar 6, 2023Updated 2 years ago
- A Python package for probabilistic state space modeling with JAX☆932Jan 6, 2026Updated last month
- Extends LasIO with Laszip integration☆12Feb 16, 2026Updated 2 weeks ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 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
- FPsolve: solver for polynomial equations over omega-continuous semirings☆11Aug 15, 2015Updated 10 years ago
- Oryx is a library for probabilistic programming and deep learning built on top of Jax.☆302Updated this week
- 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
- Factor Graph Grammars in Python☆13Jan 17, 2026Updated last month
- Fast and Differentiable Message Passing on Pairwise Markov Random Fields, ACCV 2020☆10Nov 11, 2025Updated 3 months ago
- A library implementing various linear sparse fitting techniques. Uses Tensorflow for running on GPU.☆14Dec 25, 2016Updated 9 years ago
- ☆14Feb 1, 2024Updated 2 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated last year
- ☆11Jun 17, 2016Updated 9 years ago
- Blang's software development kit☆25Nov 26, 2025Updated 3 months ago
- ☆14Jul 23, 2015Updated 10 years ago
- Map single-cell transcriptomes to copy number evolutionary trees.☆14Aug 18, 2024Updated last year
- Automatic differentiation for NumPy☆42Jun 14, 2013Updated 12 years ago
- Process airborne laser scans into raster images☆15Jan 15, 2026Updated last month
- A set of Pyro models and functions to infer CNA from scRNA-seq data☆11Aug 14, 2023Updated 2 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
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Oct 9, 2022Updated 3 years ago