/PGMax

Loopy belief propagation for factor graphs on discrete variables in JAX

Primary LanguagePythonApache License 2.0Apache-2.0

Stargazers