mctx in n2cholas/awesome-jax

Monte Carlo tree search in JAX

updated at May 19, 2024, 1:57 p.m.

29 +0

2,215 +5

172 +0

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

TensorFlow-based neural network library

updated at May 19, 2024, 5:47 a.m.

420 +0

9,691 +3

1,282 +2

alphafold in n2cholas/awesome-jax

Open source code for AlphaFold.

updated at May 19, 2024, 1:59 a.m.

224 -1

11,891 +69

2,121 +17

graph_nets in krzjoa/awesome-python-data-science

Build Graph Nets in Tensorflow

updated at May 18, 2024, 3:57 p.m.

223 +0

5,329 +4

781 +1

acme in krzjoa/awesome-python-data-science

A library of reinforcement learning components and agents

updated at May 18, 2024, 8:23 a.m.

82 +0

3,399 +11

418 +2

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

A Graph Neural Network Library in Jax

updated at May 17, 2024, 7 p.m.

41 +0

1,326 +1

87 +2

kfac-jax in n2cholas/awesome-jax

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

updated at May 17, 2024, 10:16 a.m.

10 +0

202 +0

15 +0

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

Optax is a gradient processing and optimization library for JAX.

updated at May 17, 2024, 4:20 a.m.

34 +0

1,504 +3

154 +2

chex in n2cholas/awesome-jax

None

updated at May 16, 2024, 9:03 a.m.

17 +0

722 +2

41 +1

dm_pix in n2cholas/awesome-jax

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

updated at May 16, 2024, 6:31 a.m.

10 +0

366 +1

21 +0

dm-haiku in n2cholas/awesome-jax

JAX-based neural network library

updated at May 15, 2024, 12:04 p.m.

39 +0

2,816 +0

226 +0

tf2jax in n2cholas/awesome-jax

None

updated at May 14, 2024, 8:36 p.m.

7 +0

96 +0

8 +0

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

TensorFlow Reinforcement Learning

updated at May 14, 2024, 7:06 p.m.

207 +0

3,141 +1

386 +0

einshape in n2cholas/awesome-jax

None

updated at May 13, 2024, 7:39 p.m.

7 +0

90 +1

5 +0

distrax in n2cholas/awesome-jax

None

updated at May 13, 2024, 5:01 p.m.

17 +0

512 +1

32 +0

rlax in n2cholas/awesome-jax

None

updated at May 13, 2024, 10:36 a.m.

34 +0

1,192 +2

83 -1

rc-data in seriousran/awesome-qa

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

updated at May 12, 2024, 5:51 a.m.

104 +0

1,289 +0

247 +0

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

None

updated at Jan. 4, 2024, 4:53 p.m.

8 +0

52 +0

3 +0