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

206 +2

15 +0

tf2jax in n2cholas/awesome-jax

None

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

8 +1

98 +0

8 +0

mctx in n2cholas/awesome-jax

Monte Carlo tree search in JAX

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

30 +1

2,222 +3

171 +0

einshape in n2cholas/awesome-jax

None

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

8 +1

91 +1

5 +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.

11 +1

372 +4

21 +0

alphafold in n2cholas/awesome-jax

Open source code for AlphaFold.

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

226 +2

11,958 +32

2,134 +6

distrax in n2cholas/awesome-jax

None

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

18 +1

513 +1

32 +0

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

None

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

9 +1

52 +0

3 +0

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.

42 +1

1,330 +1

87 +0

chex in n2cholas/awesome-jax

None

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

18 +1

728 +4

41 +0

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.

35 +1

1,518 +8

159 +3

acme in krzjoa/awesome-python-data-science

A library of reinforcement learning components and agents

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

83 +1

3,400 +1

418 +0

rlax in n2cholas/awesome-jax

None

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

35 +1

1,197 +3

83 +0

dm-haiku in n2cholas/awesome-jax

JAX-based neural network library

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

40 +1

2,821 +2

226 +0

graph_nets in krzjoa/awesome-python-data-science

Build Graph Nets in Tensorflow

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

224 +1

5,330 +0

781 +0

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

TensorFlow Reinforcement Learning

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

208 +1

3,139 -2

386 +0

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.

created at Dec. 20, 2017, 2:39 p.m.

24 +1

438 +1

65 +0

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.

421 +1

9,705 +7

1,285 +1

lab in josephmisiti/awesome-machine-learning

A customisable 3D platform for agent-based AI research

created at Nov. 30, 2016, 1:41 p.m.

468 +1

7,040 +4

1,364 +0

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.

105 +1

1,290 +0

243 -4