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.

9 +0

197 +2

15 +0

tf2jax in n2cholas/awesome-jax

None

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

7 +0

96 +0

8 +0

mctx in n2cholas/awesome-jax

Monte Carlo tree search in JAX

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

29 +1

2,203 +2

172 +0

einshape in n2cholas/awesome-jax

None

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

7 +0

89 -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.

10 +0

361 +8

21 -1

alphafold in n2cholas/awesome-jax

Open source code for AlphaFold.

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

219 +0

11,684 +20

2,080 +4

distrax in n2cholas/awesome-jax

None

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

17 +0

508 -1

32 +0

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

None

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

8 +0

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.

41 +0

1,318 +4

85 +0

chex in n2cholas/awesome-jax

None

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

17 +0

714 +3

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

34 +1

1,478 +8

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

82 +0

3,377 +6

415 +2

rlax in n2cholas/awesome-jax

None

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

34 +1

1,183 +5

83 +0

dm-haiku in n2cholas/awesome-jax

JAX-based neural network library

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

39 +0

2,806 -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.

223 +0

5,325 +2

780 -1

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

TensorFlow Reinforcement Learning

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

207 +0

3,138 +1

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

23 +0

432 +0

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.

420 +1

9,690 +0

1,281 +2

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

7,024 +3

1,360 +3

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.

104 +0

1,286 +0

247 +0