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

A Graph Neural Network Library in Jax

updated at May 26, 2024, 8:34 p.m.

41 +0

1,329 +3

87 +0

dm_pix in n2cholas/awesome-jax

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

updated at May 26, 2024, 7:22 p.m.

10 +0

368 +2

21 +0

alphafold in n2cholas/awesome-jax

Open source code for AlphaFold.

updated at May 26, 2024, 5:37 p.m.

224 +0

11,926 +35

2,128 +7

chex in n2cholas/awesome-jax

None

updated at May 26, 2024, 4 p.m.

17 +0

724 +2

41 +0

dm-haiku in n2cholas/awesome-jax

JAX-based neural network library

updated at May 26, 2024, 12:50 p.m.

39 +0

2,819 +3

226 +0

tf2jax in n2cholas/awesome-jax

None

updated at May 26, 2024, 10:27 a.m.

7 +0

98 +2

8 +0

graph_nets in krzjoa/awesome-python-data-science

Build Graph Nets in Tensorflow

updated at May 25, 2024, 10:04 p.m.

223 +0

5,330 +1

781 +0

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

Optax is a gradient processing and optimization library for JAX.

updated at May 25, 2024, 8:01 p.m.

34 +0

1,510 +6

156 +2

kfac-jax in n2cholas/awesome-jax

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

updated at May 25, 2024, 12:51 p.m.

10 +0

204 +2

15 +0

mctx in n2cholas/awesome-jax

Monte Carlo tree search in JAX

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

29 +0

2,219 +4

171 -1

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

TensorFlow-based neural network library

updated at May 25, 2024, 5:11 a.m.

420 +0

9,698 +7

1,284 +2

acme in krzjoa/awesome-python-data-science

A library of reinforcement learning components and agents

updated at May 24, 2024, 10:44 p.m.

82 +0

3,399 +0

418 +0

rlax in n2cholas/awesome-jax

None

updated at May 24, 2024, 12:07 p.m.

34 +0

1,194 +2

83 +0

lab in josephmisiti/awesome-machine-learning

A customisable 3D platform for agent-based AI research

updated at May 22, 2024, 6:06 p.m.

467 -1

7,036 +5

1,364 +2

narrativeqa in seriousran/awesome-qa

This repository contains the NarrativeQA dataset. It includes the list of documents with Wikipedia summaries, links to full stories, and questions and answers.

updated at May 20, 2024, 8:26 p.m.

23 +0

437 +3

65 +0

rc-data in seriousran/awesome-qa

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

updated at May 20, 2024, 7:39 a.m.

104 +0

1,290 +1

247 +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 +0

386 +0

einshape in n2cholas/awesome-jax

None

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

7 +0

90 +0

5 +0

distrax in n2cholas/awesome-jax

None

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

17 +0

512 +0

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