https://github.com/vicariousinc/PGMax
Python64
last month
Loopy belief propagation for factor graphs on discrete variables, in JAX!
MIT License