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
open_spiel
OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.learning-to-learn
Learning to Learn in TensorFlowalphageometry
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.gemma
Open weights LLM from Google DeepMind.mctx
Monte Carlo tree search in JAXcode_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.penzai
A JAX research toolkit for building, editing, and visualizing neural networks.bsuite
bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agenteducational
jraph
A Graph Neural Network Library in Jaxrc-data
Question answering dataset featured in "Teaching Machines to Read and Comprehendtapnet
Tracking Any Point (TAP)mujoco_menagerie
A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.rlax
scalable_agent
A TensorFlow implementation of Scalable Distributed Deep-RL with Importance Weighted Actor-Learner Architectures.android_env
RL research on Android devices.neural-processes
This repository contains notebook implementations of the following Neural Process variants: Conditional Neural Processes (CNPs), Neural Processes (NPs), Attentive Neural Processes (ANPs).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 Controlmaterials_discovery
xmanager
A platform for managing machine learning experimentschex
reverb
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning researchalphadev
funsearch
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.open_x_embodiment
recurrentgemma
Open weights language model from Google DeepMind, based on Griffin.ai-safety-gridworlds
This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.meltingpot
A suite of test scenarios for multi-agent reinforcement learning.dqn
Lua/Torch implementation of DQN (Nature, 2015)ithaca
Restoring and attributing ancient texts using deep neural networksdistrax
surface-distance
Library to compute surface distance based performance metrics for segmentation tasks.long-form-factuality
Benchmarking long-form factuality in large language models. Original code for our paper "Long-form factuality in large language models".tracr
alphamissense
dsprites-dataset
Dataset to assess the disentanglement properties of unsupervised learning methodsconcordia
A library for generative social simulationnarrativeqa
This repository contains the NarrativeQA dataset. It includes the list of documents with Wikipedia summaries, links to full stories, and questions and answers.clrs
lab2d
A customisable 2D platform for agent-based AI researchdqn_zoo
DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (DQN) agent.alphastar
dm_pix
PIX is an image processing library in JAX, for JAX.spriteworld
Spriteworld: a flexible, configurable python-based reinforcement learning environmentmathematics_conjectures
dm_env
A Python interface for reinforcement learning environmentsopro
official code for "Large Language Models as Optimizers"dm_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).spiral
We provide a pre-trained model for unconditional 19-step generation of CelebA-HQ imageslaunchpad
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 2018enn
gqn-datasets
Datasets used to train Generative Query Networks (GQNs) in the ‘Neural Scene Representation and Rendering’ paper.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.synjax
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
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 datasetkfac-jax
Second Order Optimization and Curvature Estimation with K-FAC in JAX.dm_alchemy
DeepMind Alchemy task environment: a meta-reinforcement learning benchmarkneural_testbed
jmp
JMP is a Mixed Precision library for JAX.perception_test
pg19
neural_networks_chomsky_hierarchy
Neural Networks and the Chomsky Hierarchyxquad
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), ICML2018xitari
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 typesmc_gradients
symplectic-gradient-adjustment
A colab that implements the Symplectic Gradient Adjustment optimizer from "The mechanics of n-player differentiable games"Love Open Source and this site? Check out how you can help us