https://github.com/vicariousinc/PGMax

Loopy belief propagation for factor graphs on discrete variables, in JAX!
https://github.com/vicariousinc/PGMax

jax python

Last synced: 3 months ago
JSON representation

Loopy belief propagation for factor graphs on discrete variables, in JAX!

• Host: GitHub
• URL: https://github.com/vicariousinc/PGMax
• Owner: vicariousinc
• License: mit
• Created: 2021-06-09T20:21:24.000Z (over 3 years ago)
• Default Branch: master
• Last Pushed: 2024-04-01T16:45:19.000Z (7 months ago)
• Last Synced: 2024-06-20T00:28:19.380Z (5 months ago)
• Topics: jax, python
• Language: Python
• Homepage: https://pgmax.readthedocs.io
• Size: 2.15 MB
• Stars: 63
• Watchers: 6
• Forks: 9
• Open Issues: 14
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

• awesome-jax - PGMax - A framework for building discrete Probabilistic Graphical Models (PGM's) and running inference inference on them via JAX. <img src="https://img.shields.io/github/stars/vicariousinc/pgmax?style=social" align="center"> (Libraries / New Libraries)

README

        
[![continuous-integration](https://github.com/vicariousinc/PGMax/actions/workflows/ci.yaml/badge.svg)](https://github.com/vicariousinc/PGMax/actions/workflows/ci.yaml)

[![PyPI version](https://badge.fury.io/py/pgmax.svg)](https://badge.fury.io/py/pgmax)

[![pre-commit.ci status](https://results.pre-commit.ci/badge/github/vicariousinc/PGMax/master.svg)](https://results.pre-commit.ci/latest/github/vicariousinc/PGMax/master)

[![codecov](https://codecov.io/gh/vicariousinc/PGMax/branch/master/graph/badge.svg?token=FrRlTDCFjk)](https://codecov.io/gh/vicariousinc/PGMax)

[![Documentation Status](https://readthedocs.org/projects/pgmax/badge/?version=latest)](https://pgmax.readthedocs.io/en/latest/?badge=latest)

# Important note: PGMax is now part of DeepMind 🎉!



Following the acquisition of Vicarious by Alphabet in 2022, we are excited to share that PGMax has moved to DeepMind! This repository is no longer being maintained, and all future developments will happen in the

new PGMax repository under DeepMind.







We look forward to the new exciting research possibilities from PGMax's integration within DeepMind's rapidly growing JAX ecosystem.