JAX-Flax-Tutorial-Image-Classification-with-Linen by 8bitmp3

How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).

created at Dec. 24, 2020, 3 a.m.

2 +0

22 +0

3 +0

jax-fid by matthias-wright

FID computation in Jax/Flax.

created at July 29, 2021, 4:05 a.m.

3 +0

21 +0

5 +0

gumbel_max_causal_gadgets_part2 by GuyLor

Part 2 of the code for the NeurIPS 2021 paper "Learning Generalized Gumbel-max Causal Mechanisms"

created at Nov. 12, 2021, 6:16 p.m.

1 +0

2 +0

0 +0

two-player-auctions by degregat

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

created at Aug. 23, 2021, 8:04 a.m.

0 +0

0 +0

0 +0