xmcgan_image_generation by google-research

None

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

4 +0

98 -1

15 +0

efax by NeilGirdhar

Exponential families for JAX

updated at May 7, 2024, 7:32 p.m.

4 +0

50 +0

3 +0

fortuna by awslabs

A Library for Uncertainty Quantification.

updated at May 7, 2024, 2:01 p.m.

12 +0

855 +0

45 +0

distrax by deepmind

None

updated at May 7, 2024, 11:22 a.m.

17 +0

511 +2

32 +0

dm_pix by deepmind

PIX is an image processing library in JAX, for JAX.

updated at May 7, 2024, 8:22 a.m.

10 +0

365 +1

21 +0

jaxlie by brentyi

Rigid transforms + Lie groups in JAX

updated at May 7, 2024, 2:07 a.m.

4 +0

202 +1

14 +0

ott by ott-jax

Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.

updated at May 6, 2024, 9:19 p.m.

10 +0

456 +0

79 +0

cr-sparse by carnotresearch

Functional models and algorithms for sparse signal processing

updated at May 5, 2024, 1:49 p.m.

6 +0

83 +0

11 +0

jax-flows by ChrisWaites

Normalizing Flows in JAX 🌊

updated at May 5, 2024, 4:30 a.m.

8 +0

263 +0

17 +0

big_transfer by google-research

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

updated at May 4, 2024, 9 a.m.

42 +0

1,493 +0

174 +0

lqg by RothkopfLab

Inverse optimal control for continuous psychophysics

updated at May 3, 2024, 2:34 p.m.

2 +0

24 +0

8 +0

lorax by davisyoshida

LoRA for arbitrary JAX models and functions

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

3 +0

117 +0

4 +0

objax by google

None

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

26 +0

761 +0

79 +0

jax-tqdm by jeremiecoullon

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

updated at May 2, 2024, 3:14 a.m.

4 +0

55 +0

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

scico by lanl

Scientific Computational Imaging COde

updated at April 30, 2024, 5:15 p.m.

8 +0

84 +0

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

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

tree-math by google

Mathematical operations for JAX pytrees

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

10 -1

167 +0

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