Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
created at Dec. 21, 2022, 9:20 a.m.
2 +0
38 +0
Image augmentation library for Jax
created at Feb. 9, 2021, 1:49 a.m.
3 +0
34 +0
FID computation in Jax/Flax.
created at July 29, 2021, 4:05 a.m.
21 +0
5 +0
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
0 +0