tensorly by tensorly

TensorLy: Tensor Learning in Python.

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

45 +0

1,558 +5

289 -1

pytorch-fid by mseitzer

Compute FID scores with PyTorch.

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

13 +0

3,383 +14

511 +2

netket by netket

Machine learning algorithms for many-body quantum systems

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

26 +0

545 +3

188 +0

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,125 +2

134,330 +301

26,858 +62

numpyro by pyro-ppl

Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

created at Feb. 13, 2019, 9:13 p.m.

53 +1

2,183 +7

239 +1

neural-tangents by google

Fast and Easy Infinite Neural Networks in Python

created at April 8, 2019, 4:48 p.m.

62 +0

2,276 +1

226 +0

jax-md by jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

created at May 13, 2019, 9:03 p.m.

47 +0

1,179 +3

195 +2

deltapv by romanodev

A photovoltaic simulator with automatic differentiation

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

6 +0

60 +0

17 +0

SymJAX by SymJAX

Documentation:

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

9 +0

120 +0

5 +0

trax by google

Trax — Deep Learning with Clear Code and Speed

created at Oct. 5, 2019, 3:09 p.m.

141 +0

8,089 +5

814 +0

cvxpylayers by cvxgrp

Differentiable convex optimization layers

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

56 +0

1,811 +8

159 +0

flax by google

Flax is a neural network library for JAX that is designed for flexibility.

created at Jan. 10, 2020, 9:48 a.m.

86 +0

6,090 +17

644 +1

jax-unirep by ElArkk

Reimplementation of the UniRep protein featurization model.

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

5 +0

104 +0

31 +0

mcx by rlouf

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

created at Jan. 22, 2020, 8:38 a.m.

17 +0

325 +1

17 +0

dm-haiku by deepmind

JAX-based neural network library

created at Feb. 18, 2020, 7:14 a.m.

37 +0

2,891 +4

233 +0

rlax by deepmind

None

created at Feb. 18, 2020, 7:14 a.m.

34 +0

1,257 +3

87 +1

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

sklearn-jax-kernels by ExpectationMax

Composable kernels for scikit-learn implemented in JAX.

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

5 +0

42 +0

4 +0

jax-flows by ChrisWaites

Normalizing Flows in JAX 🌊

created at March 23, 2020, 2:42 a.m.

8 +0

273 +0

19 +0

efax by NeilGirdhar

Exponential families for JAX

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

5 +0

55 +0

3 +0