https://github.com/vicariousinc/PGMax
Python63
5 months ago
Loopy belief propagation for factor graphs on discrete variables, in JAX!
MIT License