kalman-jax by AaltoML

Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX

created at April 3, 2020, 6:52 a.m.

11 +0

93 +0

13 +0

jax_cosmo by DifferentiableUniverseInitiative

A differentiable cosmology library in JAX

created at May 1, 2020, 5:47 p.m.

18 +0

159 +2

31 +1

big_transfer by google-research

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

created at May 11, 2020, 9:34 p.m.

42 +0

1,495 +0

174 +0

parallax by srush

None

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

6 +0

158 +1

4 +0

optax by deepmind

Optax is a gradient processing and optimization library for JAX.

created at June 12, 2020, 3:45 p.m.

34 +0

1,510 +6

156 +2

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

fourier-feature-networks by tancik

Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains

created at June 16, 2020, 11:54 p.m.

24 +0

1,194 +1

128 +1

BrainPy by brainpy

Brain Dynamics Programming in Python

created at July 16, 2020, 2:11 a.m.

12 +0

484 +2

88 +0

mpi4jax by mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python

created at July 21, 2020, 10:57 a.m.

10 +0

379 +3

26 +0

jaxns by Joshuaalbert

Probabilistic Programming and Nested sampling in JAX

created at July 31, 2020, 6 p.m.

5 +0

129 +2

8 +0

chex by deepmind

None

created at Aug. 6, 2020, 9:32 a.m.

17 +0

724 +2

41 +0

objax by google

None

created at Aug. 20, 2020, 6:20 a.m.

26 +0

761 +0

79 +0

equivariant-MLP by mfinzi

A library for programmatically generating equivariant layers through constraint solving

created at Sept. 11, 2020, 5:40 p.m.

9 +0

249 +1

19 +0

GPJax by JaxGaussianProcesses

Gaussian processes in JAX.

created at Sept. 27, 2020, 8:40 p.m.

4 +0

394 +1

47 +0

vision_transformer by google-research

None

created at Oct. 21, 2020, 12:35 p.m.

94 -1

9,460 +42

1,214 +2

vdvae by openai

Repository for the paper "Very Deep VAEs Generalize Autoregressive Models and Can Outperform Them on Images"

created at Nov. 10, 2020, 8:57 p.m.

129 +0

429 +1

84 -1

jraph by deepmind

A Graph Neural Network Library in Jax

created at Nov. 23, 2020, 10:27 a.m.

41 +0

1,329 +3

87 +0

jaxlie by brentyi

Rigid transforms + Lie groups in JAX

created at Nov. 28, 2020, 6:18 a.m.

5 +1

203 +1

14 +0

blackjax by blackjax-devs

BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.

created at Dec. 9, 2020, 8:12 a.m.

13 +0

729 +2

94 +1

exojax by HajimeKawahara

🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt

created at Dec. 12, 2020, 3:46 a.m.

4 +0

40 +0

14 +0