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