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.
Loopy belief propagation for factor graphs on discrete variables, in JAX!
updated at Jan. 4, 2024, 4:58 p.m.
Normalizing Flows using JAX
updated at March 3, 2024, 10:16 p.m.
A selection of neural network models ported from torchvision for JAX & Flax.
updated at April 11, 2024, 2:03 p.m.
Accelerated Quality-Diversity
updated at April 16, 2024, 8:40 a.m.
A differentiable cosmology library in JAX
updated at April 26, 2024, 4:33 p.m.