@article{pgmaxfactorgraphsfordiscrete,
  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 = {2022},
  eprint = {2202.04110},
  archivePrefix = {arXiv},
  url = {https://arxiv.org/abs/2202.04110v4},
}