Pinned repositories
Repositories
-
optax
Optax is a gradient processing and optimization library for JAX.
-
reverb
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
-
ferminet
An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
-
dm_alchemy
DeepMind Alchemy task environment: a meta-reinforcement learning benchmark
-
acme
A library of reinforcement learning components and agents
-
tree
tree is a library for working with nested data structures
-
open_spiel
OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
-
fancyflags
A Python library for defining flat or nested dictionary flags.
-
dm-haiku
JAX-based neural network library
-
dm_env
A Python interface for reinforcement learning environments
-
deepmind-research
This repository contains implementations and illustrative code to accompany DeepMind publications
-
sonnet
TensorFlow-based neural network library
-
dm_control
DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
-
multidim-image-augmentation Archived
This package provides TensorFlow Ops for multidimensional volumetric image augmentation.
-
bsuite
bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent
-
dm_memorytasks
A set of 13 diverse machine-learning tasks that require memory to solve.
-
lab
A customisable 3D platform for agent-based AI research
-
jraph
A Graph Neural Network Library in Jax
-
lab2d
A customisable 2D platform for agent-based AI research
-
hanabi-learning-environment
hanabi_learning_environment is a research platform for Hanabi experiments.
-
neural-processes
This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).
-
surface-distance
Library to compute surface distance based performance metrics for segmentation tasks.
-
tensor_annotations
Annotating tensor shapes using Python types
-
dsprites-dataset
Dataset to assess the disentanglement properties of unsupervised learning methods
