Loopy belief propagation for factor graphs on discrete variables, in JAX!
updated at Jan. 4, 2024, 4:58 p.m.
JAX/Haiku implementation of "Auction Learning as a Two-Player Game"
updated at Dec. 13, 2023, 10:35 a.m.
Part 2 of the code for the NeurIPS 2021 paper "Learning Generalized Gumbel-max Causal Mechanisms"
updated at Feb. 10, 2023, 9:28 a.m.