chex in n2cholas/awesome-jax

None

updated at June 9, 2024, 6:39 p.m.

18 +0

735 +7

41 +0

alphafold in n2cholas/awesome-jax

Open source code for AlphaFold.

updated at June 9, 2024, 6:06 p.m.

227 +1

11,990 +32

2,135 +1

distrax in n2cholas/awesome-jax

None

updated at June 9, 2024, 1:35 p.m.

19 +1

515 +2

32 +0

acme in krzjoa/awesome-python-data-science

A library of reinforcement learning components and agents

updated at June 9, 2024, 10:33 a.m.

83 +0

3,410 +10

418 +0

rlax in n2cholas/awesome-jax

None

updated at June 9, 2024, 3:36 a.m.

35 +0

1,203 +6

83 +0

mctx in n2cholas/awesome-jax

Monte Carlo tree search in JAX

updated at June 9, 2024, 1:52 a.m.

30 +0

2,224 +2

172 +1

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

Optax is a gradient processing and optimization library for JAX.

updated at June 8, 2024, 2:06 p.m.

36 +1

1,530 +12

161 +2

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

TensorFlow-based neural network library

updated at June 8, 2024, 1:09 p.m.

421 +0

9,711 +6

1,288 +3

dm_pix in n2cholas/awesome-jax

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

updated at June 8, 2024, 10:54 a.m.

11 +0

375 +3

21 +0

kfac-jax in n2cholas/awesome-jax

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

updated at June 7, 2024, 5:33 p.m.

11 +0

207 +1

15 +0

rc-data in seriousran/awesome-qa

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

updated at June 7, 2024, 12:45 p.m.

105 +0

1,291 +1

243 +0

dm-haiku in n2cholas/awesome-jax

JAX-based neural network library

updated at June 7, 2024, 10:25 a.m.

40 +0

2,823 +2

226 +0

graph_nets in krzjoa/awesome-python-data-science

Build Graph Nets in Tensorflow

updated at June 7, 2024, 4:01 a.m.

223 -1

5,332 +2

782 +1

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

A Graph Neural Network Library in Jax

updated at June 6, 2024, 10:19 p.m.

42 +0

1,331 +1

87 +0

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

TensorFlow Reinforcement Learning

updated at May 31, 2024, 8:44 a.m.

207 -1

3,139 +0

386 +0

einshape in n2cholas/awesome-jax

None

updated at May 29, 2024, 9:42 p.m.

8 +0

91 +0

5 +0

tf2jax in n2cholas/awesome-jax

None

updated at May 29, 2024, 1:59 p.m.

8 +0

98 +0

8 +0

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

None

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

9 +0

52 +0

3 +0