einshape in n2cholas/awesome-jax

None

created at July 22, 2021, 4:32 p.m.

7 +0

99 +1

6 +0

tf2jax in n2cholas/awesome-jax

None

created at March 2, 2022, 8:22 p.m.

7 +0

105 +1

11 +0

dm_fast_mapping in SHI-Yu-Zhe/awesome-agi-cocosci

None

created at Jan. 4, 2021, 2:15 p.m.

8 +0

54 +0

4 +0

dm_pix in n2cholas/awesome-jax

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

created at June 30, 2021, 4:25 p.m.

10 +0

389 +1

23 +1

kfac-jax in n2cholas/awesome-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

created at March 18, 2022, 10:19 a.m.

11 +0

248 +4

23 +2

chex in n2cholas/awesome-jax

None

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

17 +0

788 +3

44 +0

distrax in n2cholas/awesome-jax

None

created at April 1, 2021, 5:03 p.m.

18 +0

536 +1

32 +0

mctx in n2cholas/awesome-jax

Monte Carlo tree search in JAX

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

28 +0

2,355 +5

191 +0

rlax in n2cholas/awesome-jax

None

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

34 +0

1,264 +5

86 -1

optax in krzjoa/awesome-python-data-science, n2cholas/awesome-jax

Optax is a gradient processing and optimization library for JAX.

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

36 +0

1,697 +8

193 +3

dm-haiku in n2cholas/awesome-jax

JAX-based neural network library

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

37 +0

2,908 +14

231 -2

jraph in krzjoa/awesome-python-data-science, n2cholas/awesome-jax

A Graph Neural Network Library in Jax

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

39 +0

1,375 +1

90 +0

acme in krzjoa/awesome-python-data-science

A library of reinforcement learning components and agents

created at May 1, 2020, 9:18 a.m.

84 +0

3,514 +11

426 +1

rc-data in seriousran/awesome-qa

Question answering dataset featured in "Teaching Machines to Read and Comprehend

created at Sept. 28, 2015, 10:42 p.m.

102 +0

1,293 +0

242 +0

trfl in academic/awesome-datascience, krzjoa/awesome-python-data-science

TensorFlow Reinforcement Learning

created at Aug. 8, 2018, 2:44 p.m.

205 +0

3,134 +0

386 +0

graph_nets in krzjoa/awesome-python-data-science

Build Graph Nets in Tensorflow

created at Aug. 31, 2018, 8:19 a.m.

225 +0

5,360 +1

782 +0

alphafold in n2cholas/awesome-jax

Open source code for AlphaFold 2.

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

227 +2

12,859 +112

2,282 +21

sonnet in academic/awesome-datascience, jtoy/awesome-tensorflow, krzjoa/awesome-python-data-science

TensorFlow-based neural network library

created at April 3, 2017, 11:34 a.m.

423 +0

9,778 +6

1,299 +1