Discover google-deepmind/onetwo Open Source project

Open source code for AlphaFold.

13,132

alphafold

TensorFlow-based neural network library

12,602

sonnet

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

9,769

mujoco

StarCraft II Learning Environment

8,113

pysc2

A customisable 3D platform for agent-based AI research

8,001

lab

7,101

graph_nets

Build Graph Nets in Tensorflow

graphcast

open_spiel

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

alphageometry

learning-to-learn

Learning to Learn in TensorFlow

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

4,064

dm_control

A library of reinforcement learning components and agents

3,793

acme

TensorFlow Reinforcement Learning

3,466

trfl

JAX-based neural network library

3,136

dm-haiku

alphatensor

dnc

A TensorFlow implementation of the Differentiable Neural Computer.

Open weights LLM from Google DeepMind.

2,478

gemma

Monte Carlo tree search in JAX

2,421

mctx

code_contests

optax

Optax is a gradient processing and optimization library for JAX.

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

1,670

kinetics-i3d

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

1,639

penzai

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

1,639

mathematics_dataset

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

1,621

bsuite

educational

jraph

A Graph Neural Network Library in Jax

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

1,349

rc-data

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

1,285

mujoco_menagerie

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

1,278

tapnet

Tracking Any Point (TAP)

rlax

scalable_agent

RL research on Android devices.

981

android_env

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

977

neural-processes

Real-time behaviour synthesis with MuJoCo, using Predictive Control

969

mujoco_mpc

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

959

dramatron

tree is a library for working with nested data structures

947

tree

materials_discovery

xmanager

A platform for managing machine learning experiments

open_x_embodiment

chex

ferminet

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

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

707

reverb

funsearch

alphadev

pycolab

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

A library for generative social simulation

659

concordia

hanabi_learning_environment is a research platform for Hanabi experiments.

634

hanabi-learning-environment

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

614

recurrentgemma

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

603

ai-safety-gridworlds

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

577

meltingpot

Restoring and attributing ancient texts using deep neural networks

576

ithaca

Lua/Torch implementation of DQN (Nature, 2015)

547

dqn

Lua

546

uncertain_ground_truth

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

distrax

long-form-factuality

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

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

526

surface-distance

tracr

alphamissense

dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods

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

476

narrativeqa

clrs

lab2d

A customisable 2D platform for agent-based AI research

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

420

dqn_zoo

alphastar

dm_pix

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

official code for "Large Language Models as Optimizers"

386

opro

383

mathematics_conjectures

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

367

spriteworld

TORAX: Tokamak transport simulation in JAX

367

torax

A Python interface for reinforcement learning environments

361

dm_env

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

343

dm_robotics

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

341

spiral

launchpad

leo

Implementation of Meta-Learning with Latent Embedding Optimization

enn

streetlearn

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

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

285

gqn-datasets

An interactive HTML pretty-printer for machine learning research in IPython notebooks.

269

treescope

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

256

multi_object_datasets

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

254

AQuA

238

synjax

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

238

grid-cells

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

236

card2code

236

arnheim

Torch interface to HDF5 library

235

torch-hdf5

Lua

234

kfac-jax

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

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

231

dm_memorytasks

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

221

Temporal-3D-Pose-Kinetics

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark

218

dm_alchemy

neural_testbed

perception_test

jmp

JMP is a Mixed Precision library for JAX.

Neural Networks and the Chomsky Hierarchy

183

neural_networks_chomsky_hierarchy

xquad

nanodo

pg19

spectral_inference_networks

Implementation of Spectral Inference Networks, ICLR 2019

Barkour Robot: Agile Quadruped Robots by Google DeepMind

165

barkour_robot