jax-unirep by ElArkk

Reimplementation of the UniRep protein featurization model.

created at Jan. 21, 2020, 3:30 p.m.

6 +0

101 +0

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

SymJAX by SymJAX

Documentation:

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

8 +0

117 -1

5 +0

ott by google-research

None

created at Jan. 18, 2021, 3:03 p.m.

15 +0

215 +0

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

tensorly by tensorly

TensorLy: Tensor Learning in Python.

created at Oct. 21, 2016, 11:14 p.m.

45 +0

1,504 +0

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

cvxpylayers by cvxgrp

Differentiable convex optimization layers

created at Oct. 27, 2019, 10:42 p.m.

53 +0

1,730 +7

154 +0

coax by coax-dev

Modular framework for Reinforcement Learning in python

created at March 11, 2021, 12:56 a.m.

8 +0

161 +0

18 +0

deltapv by romanodev

A photovoltaic simulator with automatic differentiation

created at May 20, 2019, 8:23 p.m.

6 +0

53 +0

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

cr-sparse by carnotresearch

Functional models and algorithms for sparse signal processing

created at Dec. 22, 2020, 8:31 a.m.

6 +0

85 +1

11 +0

brax by google

Massively parallel rigidbody physics simulation on accelerator hardware.

created at June 2, 2021, 9:45 p.m.

36 +0

2,090 +7

231 +1

flaxmodels by matthias-wright

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

created at May 16, 2021, 1:34 p.m.

6 +0

223 +0

24 +0

jaxrl by ikostrikov

JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.

created at Jan. 16, 2021, 1:26 a.m.

12 +0

588 +4

62 +0

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

jaxopt by google

Hardware accelerated, batchable and differentiable optimizers in JAX.

created at July 12, 2021, 5:16 p.m.

19 +0

895 +1

60 +0

netket by netket

Machine learning algorithms for many-body quantum systems

created at April 23, 2018, 6:48 p.m.

25 +0

505 +2

174 -1

pytorch-fid by mseitzer

Compute FID scores with PyTorch.

created at Feb. 10, 2018, 12:33 p.m.

14 +0

3,146 +15

492 +1

transformers by huggingface

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

created at Oct. 29, 2018, 1:56 p.m.

1,101 +4

126,623 +368

25,069 +70