Loopy belief propagation for factor graphs on discrete variables, in JAX!
☆66Oct 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☆167Oct 18, 2024Updated last year
- Probabilistic inference for models of behaviour☆13Mar 5, 2026Updated 3 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
- Tools for the symbolic manipulation of PyMC models, Theano, and TensorFlow graphs.☆65Mar 20, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 4 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
- 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
- The uncompromising Jupyter notebook formatter.☆14Jan 26, 2024Updated 2 years ago
- Documentation:☆131May 22, 2023Updated 3 years ago
- Sparse nonlinear least squares in JAX☆287Updated this week
- Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.☆332Mar 20, 2024Updated 2 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.
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 11 months ago
- E.T. Jaynes home phone.☆45Jan 11, 2021Updated 5 years ago
- Blang's software development kit☆25Nov 26, 2025Updated 6 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☆62Jun 17, 2024Updated last year
- Probabilistic programming system for fast and exact symbolic inference☆86Apr 12, 2026Updated 2 months ago
- Automatic differentiation for NumPy☆42Jun 14, 2013Updated 13 years ago
- A Python package for probabilistic state space modeling with JAX☆970Jan 6, 2026Updated 5 months ago
- Jax SSM Library☆48Nov 24, 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.
- 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.☆317Jun 4, 2026Updated last week
- A Scalable Approximate Method for Probabilistic Neurosymbolic Inference☆25Jan 27, 2025Updated last year
- Finding the causality in biological pathways☆14Jul 12, 2023Updated 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
- Automatic Reparameterisation of Probabilistic Programs☆36Jun 11, 2020Updated 6 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
- 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
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Movies Recommendation with Hierarchical Poisson Factorization in Edward☆18Jun 20, 2017Updated 8 years ago
- Sparse Signaling Pathway Sampling: MCMC for signaling pathway inference☆14Oct 24, 2022Updated 3 years ago
- Agent models implemented with Pyro☆11Jul 11, 2023Updated 2 years ago
- ☆635May 13, 2026Updated last month
- Julia implementation of elliptical slice sampling.☆16May 26, 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
- Learning Deep Parameterized Skills from Demonstration for Re-targetable Visuomotor Control☆17Apr 8, 2020Updated 6 years ago