sonnet

TensorFlow-based neural network library

Creator

google-deepmind

Related apps

acme

acme

A library of reinforcement learning components and agents

Python3269apache-2.0

5 months ago

agentsreinforcement-learningresearch

distrax

Python490apache-2.0

4 months ago

dm_pix

dm_pix

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

Python346apache-2.0

2 months ago

computer-visionimageimage-processing

einshape

einshape

Python90apache-2.0

last year

graph_nets

graph_nets

Build Graph Nets in Tensorflow

Python5303apache-2.0

last year

artificial-intelligencedeep-learninggraph-networks

lab

lab

A customisable 3D platform for agent-based AI research

C6959other

last year

artificial-intelligencedeep-learningmachine-learning

mctx

Monte Carlo tree search in JAX

Python2162apache-2.0

3 months ago

jaxmonte-carlo-tree-searchreinforcement-learning

rc-data

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

Python1287apache-2.0

7 years ago

tf2jax

Python93apache-2.0

3 months ago

trfl

TensorFlow Reinforcement Learning

Python3140apache-2.0

last year

chex

Python682apache-2.0

2 months ago

dm-haiku

JAX-based neural network library

Python2715apache-2.0

2 months ago

deep-learningdeep-neural-networksjax

jraph

jraph

A Graph Neural Network Library in Jax

Python1267apache-2.0

last year

deep-learninggraph-neural-networksjax

rlax

Python1149apache-2.0

5 months ago

alphafold

alphafold

Open source code for AlphaFold.

Python11209apache-2.0

5 months ago

kfac-jax

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

Python186apache-2.0

2 months ago

bayesian-deep-learningmachine-learningoptimization

narrativeqa

This repository contains the NarrativeQA dataset. It includes the list of docume

Shell431apache-2.0

4 years ago

torch-cephes

Cephes Mathematical Functions library wrapped for Torch

C47other

8 years ago

torch-randomkit

Provides and wraps the Randomkit library, copied from Numpy.

Lua34bsd-3-clause

5 years ago