PGMax Reference Documentation¶
PGMax is a library for working with Factor Graphs in JAX. It currently provides an interface for specifying factor graphs of any type, as well as an efficient implementation of max-product belief propagation and inference on these graphs.
Contribute¶
Issue tracker: https://github.com/deepmind/PGMax/issues
Source code: https://github.com/deepmind/PGMax/tree/main
Support¶
If you are having issues, please let us know by filing an issue on our issue tracker.
License¶
PGMax is licensed under the Apache 2.0 License.