Robuta

https://github.com/vicariousinc/PGMax GitHub - vicariousinc/PGMax: Loopy belief propagation for factor graphs on discrete variables, in... Loopy belief propagation for factor graphs on discrete variables, in JAX! - vicariousinc/PGMax