GitHub repository: vicariousinc/PGMax
Primary language: Python
Description: Loopy belief propagation for factor graphs on discrete variables, in JAX!