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.

Getting Started

• Installation Guide

API Documentation

• PGMax API reference
  • Factor
    • factor
    • enum
    • logical
    • pool
  • Fgraph
    • fgraph
  • Fgroup
    • fgroup
    • enum
    • logical
    • pool
  • Infer
    • bp
    • bp_state
    • dual_lp
    • energy
    • inferer
  • Vgroup
    • vgroup
    • varray
    • vdict
  • Utils

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.

Indices and tables

• Index

• Module Index

• Search Page