There are no reviews yet. Be the first to send feedback to the community and the maintainers!
deepmind-research
This repository contains implementations and illustrative code to accompany DeepMind publicationsalphafold
Open source code for AlphaFold.sonnet
TensorFlow-based neural network librarypysc2
StarCraft II Learning Environmentmujoco
Multi-Joint dynamics with Contact. A general purpose physics simulator.lab
A customisable 3D platform for agent-based AI researchgraph_nets
Build Graph Nets in Tensorflowgraphcast
learning-to-learn
Learning to Learn in TensorFlowopen_spiel
OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.alphageometry
dm_control
Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.acme
A library of reinforcement learning components and agentstrfl
TensorFlow Reinforcement Learningdm-haiku
JAX-based neural network libraryalphatensor
dnc
A TensorFlow implementation of the Differentiable Neural Computer.mctx
Monte Carlo tree search in JAXgemma
Open weights LLM from Google DeepMind.code_contests
kinetics-i3d
Convolutional neural network model for video classification trained on the Kinetics dataset.mathematics_dataset
This dataset code generates mathematical question and answer pairs, from a range of question types at roughly school-level difficulty.optax
Optax is a gradient processing and optimization library for JAX.bsuite
bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agentpenzai
A JAX research toolkit for building, editing, and visualizing neural networks.educational
jraph
A Graph Neural Network Library in Jaxrc-data
Question answering dataset featured in "Teaching Machines to Read and Comprehendrlax
tapnet
Tracking Any Point (TAP)scalable_agent
A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.neural-processes
This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).android_env
RL research on Android devices.mujoco_menagerie
A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.dramatron
Dramatron uses large language models to generate coherent scripts and screenplays.tree
tree is a library for working with nested data structuresmujoco_mpc
Real-time behaviour synthesis with MuJoCo, using Predictive Controlxmanager
A platform for managing machine learning experimentsmaterials_discovery
chex
reverb
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning researchalphadev
pycolab
A highly-customisable gridworld game engine with some batteries included. Make your own gridworld games to test reinforcement learning agents!ferminet
An implementation of the Fermionic Neural Network for ab-initio electronic structure calculationshanabi-learning-environment
hanabi_learning_environment is a research platform for Hanabi experiments.funsearch
ai-safety-gridworlds
This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.dqn
Lua/Torch implementation of DQN (Nature, 2015)ithaca
Restoring and attributing ancient texts using deep neural networksmeltingpot
A suite of test scenarios for multi-agent reinforcement learning.distrax
surface-distance
Library to compute surface distance based performance metrics for segmentation tasks.tracr
dsprites-dataset
Dataset to assess the disentanglement properties of unsupervised learning methodsalphamissense
long-form-factuality
Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".narrativeqa
This repository contains the NarrativeQA dataset. It includes the list of documents with Wikipedia summaries, links to full stories, and questions and answers.lab2d
A customisable 2D platform for agent-based AI researchopen_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 environmentdm_pix
PIX is an image processing library in JAX, for JAX.concordia
A library for generative social simulationmathematics_conjectures
alphastar
spiral
We provide a pre-trained model for unconditional 19-step generation of CelebA-HQ imagesdm_env
A Python interface for reinforcement learning environmentsdm_robotics
Libraries, tools and tasks created and used at DeepMind Robotics.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).launchpad
leo
Implementation of Meta-Learning with Latent Embedding Optimizationstreetlearn
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 2018gqn-datasets
Datasets used to train Generative Query Networks (GQNs) in the ‘Neural Scene Representation and Rendering’ paper.enn
opro
official code for "Large Language Models as Optimizers"multi_object_datasets
Multi-object image datasets with ground-truth segmentation masks and generative factors.AQuA
A algebraic word problem dataset, with multiple choice questions annotated with rationales.card2code
A code generation dataset for generating the code that implements Hearthstone and Magic The Gathering card effects.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-6arnheim
synjax
torch-hdf5
Torch interface to HDF5 librarydm_memorytasks
A set of 13 diverse machine-learning tasks that require memory to solve.Temporal-3D-Pose-Kinetics
Exploiting temporal context for 3D human pose estimation in the wild: 3D poses for the Kinetics datasetdm_alchemy
DeepMind Alchemy task environment: a meta-reinforcement learning benchmarkneural_testbed
kfac-jax
Second Order Optimization and Curvature Estimation with K-FAC in JAX.pg19
jmp
JMP is a Mixed Precision library for JAX.xquad
spectral_inference_networks
Implementation of Spectral Inference Networks, ICLR 2019abstract-reasoning-matrices
Progressive matrices dataset, as described in: Measuring abstract reasoning in neural networks (Barrett*, Hill*, Santoro*, Morcos, Lillicrap), ICML2018neural_networks_chomsky_hierarchy
Neural Networks and the Chomsky Hierarchyxitari
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.tensor_annotations
Annotating tensor shapes using Python typessymplectic-gradient-adjustment
A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"mc_gradients
interval-bound-propagation
This repository contains a simple implementation of Interval Bound Propagation (IBP) using TensorFlow: https://arxiv.org/abs/1810.12715s6
Love Open Source and this site? Check out how you can help us