@misc{indiciaebecab67aa404, title = {PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX}, author = {Guangyao Zhou and Antoine Dedieu and Nishanth Kumar and Wolfgang Lehrach and Miguel Lázaro-Gredilla and Shrinu Kushagra and Dileep George}, year = {2023}, url = {https://arxiv.org/abs/2202.04110}, note = {Source identifier: 2202.04110} }