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

alphafold by deepmind

Open source code for AlphaFold.

created at June 17, 2021, 2:06 p.m.

224 +0

11,926 +35

2,128 +7

trax by google

Trax — Deep Learning with Clear Code and Speed

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

146 +0

7,969 +6

811 +0

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

vision_transformer by google-research

None

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

94 -1

9,460 +42

1,214 +2

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.

84 +1

5,596 +26

595 +2

neural-tangents by google

Fast and Easy Infinite Neural Networks in Python

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

64 +0

2,238 +4

226 +0

cvxpylayers by cvxgrp

Differentiable convex optimization layers

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

53 +0

1,730 +7

154 +0

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.

49 +0

2,062 +5

222 +2

jax-md by jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

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

47 +0

1,106 +3

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

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

EasyLM by young-geng

Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Flax.

created at Nov. 22, 2022, 12:55 p.m.

41 +0

2,257 +9

233 +0

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

dm-haiku by deepmind

JAX-based neural network library

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

39 +0

2,819 +3

226 +0

scenic by google-research

Scenic: A Jax Library for Computer Vision Research and Beyond

created at July 12, 2021, 2:27 p.m.

39 +0

3,046 +10

407 +3

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

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

rlax by deepmind

None

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

34 +0

1,194 +2

83 +0

mctx by deepmind

Monte Carlo tree search in JAX

created at March 1, 2022, 5:26 p.m.

29 +0

2,219 +4

171 -1