Skip to content
OpenTrain AIFor AI Companies

PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX

Published Feb 1, 2022
arXiv PDF
Researcher verdict
Starting point
Use as implementation starting point
Benchmark evidence
Missing
Not verified yet
Time to first repro
A few hours
Fast first run
Risk flags
0
None detected

Results and benchmarks

Freshness tier: cold
PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX is the primary contribution described in this paper.

Implementation

Best maintained implementation now

Recommended
Confidence: High
Reproducibility: Strong
deepmind/pgmax

Loopy belief propagation for factor graphs on discrete variables in JAX

166 stars · 12 forks · Last push Oct 18, 2024 · Apache-2.0 license

  • License
  • CI
  • Dependencies
  • Docker

Official implementation from Papers with Code · Repository link is mentioned in the paper metadata · Strong overlap with paper title keywords

Why this implementation
Confidence: high

deepmind/pgmax is the strongest maintained implementation based on ranking signals. CI workflows are present. License is declared (Apache-2.0).

Open deepmind/pgmax
Reproduction risks
  • No repository-level red flags were detected, but paper-specific preprocessing and hyperparameter details may still be under-specified.
  • Selected deepmind/pgmax as the strongest maintained implementation for new work.
  • Includes CI workflow signals.
  • Includes dependency/environment manifest signals.
  • Repository activity is within the last 24 months.

Compare implementation paths

Compare maintenance quality, reproducibility coverage, and evidence confidence before choosing a reproduction baseline.

deepmind/pgmax
best maintained
Maintenance
Stale
Confidence
High
Reproducibility
Strong
Stars
166
Last push
Oct 18, 2024 (677d)

Official implementation from Papers with Code · Repository link is mentioned in the paper metadata

  • No push in 12+ months
  • No Docker setup
vicariousinc/pgmax
historical official
Maintenance
Archived
Confidence
High
Reproducibility
Limited
Stars
66
Last push
Oct 7, 2024 (687d)

Official implementation from Papers with Code · Repository link is mentioned in the paper metadata

  • Repository archived
  • No push in 12+ months
  • No Docker setup
google-deepmind/PGMax
alternative
Maintenance
Stale
Confidence
Low
Reproducibility
Strong
Stars
166
Last push
Oct 18, 2024 (677d)

Strong overlap with paper title keywords · Community adoption signal (166 stars)

  • No push in 12+ months
  • No Docker setup
  • Low confidence match

Reproduction readiness

Time to first repro: hours
Last checked: Aug 22, 2026

Setup required

Dependencies pinned, manual setup needed

  • deepmind/pgmax has requirements.txt but requires manual environment setup.
  • Last push was 677 days ago, so expect possible dependency version conflicts.
  • No Dockerfile, so you will set up the environment manually.
Open deepmind/pgmax

Quick start

git clone https://github.com/deepmind/pgmax.git
pip install -r requirements.txt

Repositories and ecosystem

No additional verified repositories beyond the primary recommendation.

Hugging Face artifacts

No direct paper-linked artifacts were found. Showing strongest curated related artifacts for faster exploration.

Models

No trustworthy models matches right now.

Search models on Hugging Face

Datasets

Curated Related

Broaden dataset search

pgmax factor graphs discrete dataset

Spaces

No trustworthy spaces matches right now.

Search spaces on Hugging Face

Research context

Evaluation and human feedback data

Open this paper in HFEPX to review benchmark signals, evaluation modes, and human-feedback protocol context.

Open in HFEPX

Data includes links from Papers with Code ( CC-BY-SA-4.0 ).