Explore @google-deepmind Open Source projects

Open source code for AlphaFold.

12,791

alphafold

TensorFlow-based neural network library

11,697

sonnet

StarCraft II Learning Environment

9,682

pysc2

Multi-Joint dynamics with Contact. A general purpose physics simulator.

7,904

mujoco

A customisable 3D platform for agent-based AI research

7,069

lab

7,012

graph_nets

Build Graph Nets in Tensorflow

graphcast

learning-to-learn

Learning to Learn in TensorFlow

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

4,063

open_spiel

alphageometry

dm_control

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

A library of reinforcement learning components and agents

3,473

acme

TensorFlow Reinforcement Learning

3,372

trfl

JAX-based neural network library

3,139

dm-haiku

alphatensor

dnc

A TensorFlow implementation of the Differentiable Neural Computer.

Monte Carlo tree search in JAX

2,478

mctx

Open weights LLM from Google DeepMind.

2,179

gemma

code_contests

kinetics-i3d

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.

1,639

mathematics_dataset

Optax is a gradient processing and optimization library for JAX.

1,621

optax

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

1,486

bsuite

A JAX research toolkit for building, editing, and visualizing neural networks.

1,465

penzai

educational

jraph

A Graph Neural Network Library in Jax

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

1,306

rc-data

rlax

tapnet

Tracking Any Point (TAP)

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

1,033

scalable_agent

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

972

neural-processes

RL research on Android devices.

966

android_env

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.

946

mujoco_menagerie

Dramatron uses large language models to generate coherent scripts and screenplays.

926

dramatron

tree is a library for working with nested data structures

904

tree

A platform for managing machine learning experiments

891

xmanager

Real-time behaviour synthesis with MuJoCo, using Predictive Control

794

mujoco_mpc

materials_discovery

chex

reverb

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

alphadev

pycolab

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

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

654

ferminet

hanabi_learning_environment is a research platform for Hanabi experiments.

643

hanabi-learning-environment

funsearch

ai-safety-gridworlds

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

Lua/Torch implementation of DQN (Nature, 2015)

577

dqn

Lua

546

ithaca

Restoring and attributing ancient texts using deep neural networks

A suite of test scenarios for multi-agent reinforcement learning.

535

meltingpot

Open weights language model from Google DeepMind, based on Griffin.

516

recurrentgemma

distrax

surface-distance

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

Dataset to assess the disentanglement properties of unsupervised learning methods

493

dsprites-dataset

tracr

alphamissense

narrativeqa

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

Shell

432

lab2d

A customisable 2D platform for agent-based AI research

open_x_embodiment

dqn_zoo

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

clrs

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment

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

366

dm_pix

A library for generative social simulation

363

concordia

We provide a pre-trained model for unconditional 19-step generation of CelebA-HQ images

351

mathematics_conjectures

alphastar

spiral

A Python interface for reinforcement learning environments

327

dm_env

Libraries, tools and tasks created and used at DeepMind Robotics.

326

dm_robotics

Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggregation from our recent work on uncertain ground truth (TMLR'23 and ArXiv pre-print).

315

uncertain_ground_truth

Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".

315

long-form-factuality

launchpad

leo

Implementation of Meta-Learning with Latent Embedding Optimization

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

299

streetlearn

Datasets used to train Generative Query Networks (GQNs) in the ‘Neural Scene Representation and Rendering’ paper.

279

gqn-datasets

enn

multi_object_datasets

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

A algebraic word problem dataset, with multiple choice questions annotated with rationales.

247

AQuA

238

card2code

A code generation dataset for generating the code that implements Hearthstone and Magic The Gathering card effects.

236

grid-cells

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

arnheim

synjax

torch-hdf5

Torch interface to HDF5 library

Lua

231

dm_memorytasks

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

220

Temporal-3D-Pose-Kinetics

official code for "Large Language Models as Optimizers"

214

opro

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark

199

dm_alchemy

neural_testbed

kfac-jax

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

pg19

xquad

jmp

JMP is a Mixed Precision library for JAX.

Implementation of Spectral Inference Networks, ICLR 2019

171

spectral_inference_networks

Progressive matrices dataset, as described in: Measuring abstract reasoning in neural networks (Barrett*, Hill*, Santoro*, Morcos, Lillicrap), ICML2018

165

abstract-reasoning-matrices

162

xitari

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.

Annotating tensor shapes using Python types

159

tensor_annotations

Neural Networks and the Chomsky Hierarchy

159

neural_networks_chomsky_hierarchy

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

155

symplectic-gradient-adjustment

mc_gradients

interval-bound-propagation

This repository contains a simple implementation of Interval Bound Propagation (IBP) using TensorFlow: https://arxiv.org/abs/1810.12715