gumbel_max_causal_gadgets_part2 by GuyLor

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.

1 +0

2 +0

0 +0

PGMax by vicariousinc

Loopy belief propagation for factor graphs on discrete variables, in JAX!

updated at Jan. 4, 2024, 4:58 p.m.

6 +0

63 +0

9 +0

jax-fid by matthias-wright

FID computation in Jax/Flax.

updated at Jan. 4, 2024, 4:59 p.m.

3 +0

21 +0

5 +0

NuX by Information-Fusion-Lab-Umass

Normalizing Flows using JAX

updated at March 3, 2024, 10:16 p.m.

9 +0

82 +0

4 +0

deltapv by romanodev

A photovoltaic simulator with automatic differentiation

updated at March 27, 2024, 8:16 a.m.

6 +0

53 +0

16 +0

imax by 4rtemi5

Image augmentation library for Jax

updated at March 31, 2024, 1:22 a.m.

3 +0

34 +0

3 +0

flaxvision by rolandgvc

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

updated at April 11, 2024, 2:03 p.m.

4 +0

44 +0

2 +0

QDax by adaptive-intelligent-robotics

Accelerated Quality-Diversity

updated at April 16, 2024, 8:40 a.m.

6 +0

243 +0

36 +0

ott by google-research

None

updated at April 21, 2024, 10:34 p.m.

15 +0

215 +0

18 +0

mcx by rlouf

Express & compile probabilistic programs for performant inference on CPU & GPU. Powered by JAX.

updated at April 25, 2024, 1:56 a.m.

17 +0

324 +0

17 +0

coax by coax-dev

Modular framework for Reinforcement Learning in python

updated at April 27, 2024, 8:39 p.m.

8 +0

161 +0

18 +0

jaxdf by ucl-bug

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

updated at April 27, 2024, 9:07 p.m.

7 +0

107 +0

7 +0

jaxfit by Dipolar-Quantum-Gases

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

updated at April 27, 2024, 9:13 p.m.

1 +0

43 +0

3 +0

jax-models by DarshanDeshpande

Unofficial JAX implementations of deep learning research papers

updated at April 29, 2024, 12:39 a.m.

5 +0

140 +0

9 +0

safejax by alvarobartt

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

updated at May 1, 2024, 8:11 p.m.

2 +0

38 +0

2 +0

objax by google

None

updated at May 2, 2024, 11:59 a.m.

26 +0

761 +0

79 +0

xmcgan_image_generation by google-research

None

updated at May 8, 2024, 4:32 a.m.

4 +0

98 +0

15 +0

bayex by alonfnt

Minimal Implementation of Bayesian Optimization in JAX

updated at May 8, 2024, 11:15 a.m.

3 +0

78 +0

1 +0

eqxvision by paganpasta

A Python package of computer vision models for the Equinox ecosystem.

updated at May 8, 2024, 8:14 p.m.

4 +0

95 +0

10 +0

flaxmodels by matthias-wright

Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.

updated at May 9, 2024, 12:33 p.m.

6 +0

223 +0

24 +0