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.
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.
JAX/Haiku implementation of "Auction Learning as a Two-Player Game"
created at Aug. 23, 2021, 8:04 a.m.