jax-fid by matthias-wright

FID computation in Jax/Flax.

updated at Jan. 4, 2024, 4:59 p.m.

3 +0

21 +0

5 +0

PGMax by vicariousinc

Loopy belief propagation for factor graphs on discrete variables, in JAX!

updated at Jan. 4, 2024, 4:58 p.m.

6 +0

63 +0

9 +0

two-player-auctions by degregat

JAX/Haiku implementation of "Auction Learning as a Two-Player Game"

updated at Dec. 13, 2023, 10:35 a.m.

0 +0

0 +0

0 +0

gumbel_max_causal_gadgets_part2 by GuyLor

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.

1 +0

2 +0

0 +0