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

bayex by alonfnt

Minimal Implementation of Bayesian Optimization in JAX

created at Aug. 1, 2021, 8:41 a.m.

3 +0

78 +0

1 +0

flaxvision by rolandgvc

A selection of neural network models ported from torchvision for JAX & Flax.

created at June 14, 2020, 4:34 p.m.

4 +0

44 +0

2 +0

jaxfit by Dipolar-Quantum-Gases

GPU/TPU accelerated nonlinear least-squares curve fitting using JAX

created at Aug. 8, 2022, 5 p.m.

1 +0

43 +0

2 +0

safejax by alvarobartt

Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`

created at Dec. 21, 2022, 9:20 a.m.

2 +0

38 +0

2 +0

GNN-for-Combinatorial-Optimization by IvanIsCoding

JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.

created at Jan. 4, 2023, 11:09 p.m.

3 +0

38 +0

2 +0

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

imax by 4rtemi5

Image augmentation library for Jax

created at Feb. 9, 2021, 1:49 a.m.

3 +0

34 +0

3 +0

efax by NeilGirdhar

Exponential families for JAX

created at March 24, 2020, 5:44 a.m.

4 +0

50 +1

3 +0

kernex by ASEM000

Stencil computations in JAX

created at July 10, 2022, 10:01 a.m.

1 +0

59 +0

3 +0

NuX by Information-Fusion-Lab-Umass

Normalizing Flows using JAX

created at March 9, 2020, 12:38 a.m.

9 +0

82 +0

4 +0

lorax by davisyoshida

LoRA for arbitrary JAX models and functions

created at April 21, 2023, 11:50 a.m.

3 +0

117 +2

4 +0

jax-tqdm by jeremiecoullon

Add a tqdm progress bar to your JAX scans and loops.

created at Jan. 16, 2023, 4:54 p.m.

4 +0

55 +1

4 +0

parallax by srush

None

created at May 19, 2020, 1:49 a.m.

6 +0

157 +0

4 +0

sklearn-jax-kernels by ExpectationMax

Composable kernels for scikit-learn implemented in JAX.

created at March 10, 2020, 6:59 p.m.

5 +0

41 +0

4 +0

SymJAX by SymJAX

Documentation:

created at Sept. 4, 2019, 2:23 p.m.

8 +0

117 +0

5 +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

einshape by deepmind

None

created at July 22, 2021, 4:32 p.m.

7 +0

89 +0

5 +0

jaxdf by ucl-bug

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations

created at Sept. 8, 2021, 4:38 p.m.

7 +0

107 +0

7 +0