90 open source projects by deepmind

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

Torch interface to HDF5 library

A TensorFlow implementation of the Differentiable Neural Computer.

A C++/Python implementation of the StreetLearn environment based on images from Street View, as well as a TensorFlow implementation of goal-driven navigation agents solving the task published in “Learning to Navigate in Cities Without a Map”, NeurIPS 2018

Library to compute surface distance based performance metrics for segmentation tasks.

DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (DQN) agent.

DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

A set of 13 diverse machine-learning tasks that require memory to solve.

Exploiting temporal context for 3D human pose estimation in the wild: 3D poses for the Kinetics dataset

A library of reinforcement learning components and agents

Implementation of the supervised learning experiments in Vector-based navigation using grid-like representations in artificial agents, as published at https://www.nature.com/articles/s41586-018-0102-6

Implementation of Spectral Inference Networks, ICLR 2019

This is the 0.4 release of the Arcade Learning Environment (ALE), a platform designed for AI research. ALE is based on Stella, an Atari 2600 VCS emulator.

A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"

A Python interface for reinforcement learning environments

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark

Multi-object image datasets with ground-truth segmentation masks and generative factors.

A class system for Lua.

This package provides TensorFlow Ops for multidimensional volumetric image augmentation.

Convolutional neural network model for video classification trained on the Kinetics dataset.

This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.

TensorFlow-based neural network library

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

This repository contains the 3D shapes dataset, used in Kim, Hyunjik and Mnih, Andriy. "Disentangling by Factorising." In Proceedings of the 35th International Conference on Machine Learning (ICML). 2018. to assess the disentanglement properties of unsupervised learning methods.

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent

Neural network verification in JAX

Probability distributions, wrapped for Torch.

Annotating tensor shapes using Python types

JAX-based neural network library

Datasets used to train Spatial Language Integrating Model (SLIM) in the ‘Encoding Spatial Relations from Natural Language’ paper.

A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.

LuaJIT wrapper for PLplot

Disclaimer: This is not an official Google product.

This repository contains implementations and illustrative code to accompany DeepMind publications

This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).

A highly-customisable gridworld game engine with some batteries included. Make your own gridworld games to test reinforcement learning agents!

tree is a library for working with nested data structures

Build Graph Nets in Tensorflow

hanabi_learning_environment is a research platform for Hanabi experiments.

A Graph Neural Network Library in Jax

This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.

Lua/Torch implementation of DQN (Nature, 2015)

Learning to Learn in TensorFlow

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

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations