TY - RPRT TI - PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX AU - Guangyao Zhou AU - Antoine Dedieu AU - Nishanth Kumar AU - Wolfgang Lehrach AU - Miguel Lázaro-Gredilla AU - Shrinu Kushagra AU - Dileep George PY - 2023 UR - https://arxiv.org/abs/2202.04110 ID - 2202.04110 ER -