DeepMind (@google-deepmind)
  • Stars
    star
    147,371
  • Global Org. Rank 66 (Top 0.03 %)
  • Registered almost 10 years ago
  • Most used languages
    Python
    67.5 %
    C++
    4.8 %
    Lua
    4.8 %
    Go
    1.3 %
    C
    1.3 %
    Others
    2.8 %

Top repositories

1

deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications
Jupyter Notebook
12,892
star
2

alphafold

Open source code for AlphaFold.
Python
12,006
star
3

sonnet

TensorFlow-based neural network library
Python
9,711
star
4

pysc2

StarCraft II Learning Environment
Python
7,904
star
5

mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.
Jupyter Notebook
7,385
star
6

lab

A customisable 3D platform for agent-based AI research
C
7,037
star
7

graph_nets

Build Graph Nets in Tensorflow
Python
5,325
star
8

graphcast

Python
4,242
star
9

learning-to-learn

Learning to Learn in TensorFlow
Python
4,063
star
10

open_spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
C++
4,060
star
11

dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.
Python
3,601
star
12

alphageometry

Python
3,580
star
13

acme

A library of reinforcement learning components and agents
Python
3,410
star
14

trfl

TensorFlow Reinforcement Learning
Python
3,139
star
15

dm-haiku

JAX-based neural network library
Python
2,816
star
16

alphatensor

Python
2,616
star
17

dnc

A TensorFlow implementation of the Differentiable Neural Computer.
Python
2,478
star
18

mctx

Monte Carlo tree search in JAX
Python
2,209
star
19

gemma

Open weights LLM from Google DeepMind.
Jupyter Notebook
2,061
star
20

code_contests

C++
2,010
star
21

kinetics-i3d

Convolutional neural network model for video classification trained on the Kinetics dataset.
Python
1,639
star
22

mathematics_dataset

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

optax

Optax is a gradient processing and optimization library for JAX.
Python
1,520
star
24

bsuite

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent
Python
1,478
star
25

penzai

A JAX research toolkit for building, editing, and visualizing neural networks.
Python
1,405
star
26

educational

Jupyter Notebook
1,386
star
27

jraph

A Graph Neural Network Library in Jax
Python
1,330
star
28

rc-data

Question answering dataset featured in "Teaching Machines to Read and Comprehend
Python
1,285
star
29

rlax

Python
1,192
star
30

tapnet

Tracking Any Point (TAP)
Python
1,122
star
31

mujoco_menagerie

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.
Jupyter Notebook
1,060
star
32

scalable_agent

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

neural-processes

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

android_env

RL research on Android devices.
Python
957
star
35

dramatron

Dramatron uses large language models to generate coherent scripts and screenplays.
Jupyter Notebook
924
star
36

tree

tree is a library for working with nested data structures
Python
908
star
37

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control
C++
830
star
38

materials_discovery

Python
825
star
39

xmanager

A platform for managing machine learning experiments
Python
800
star
40

chex

Python
722
star
41

reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research
C++
693
star
42

alphadev

Python
662
star
43

pycolab

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

ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations
Python
651
star
45

hanabi-learning-environment

hanabi_learning_environment is a research platform for Hanabi experiments.
Python
614
star
46

funsearch

Jupyter Notebook
611
star
47

ai-safety-gridworlds

This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.
Python
577
star
48

meltingpot

A suite of test scenarios for multi-agent reinforcement learning.
Python
547
star
49

dqn

Lua/Torch implementation of DQN (Nature, 2015)
Lua
546
star
50

recurrentgemma

Open weights language model from Google DeepMind, based on Griffin.
Python
544
star
51

ithaca

Restoring and attributing ancient texts using deep neural networks
Jupyter Notebook
540
star
52

distrax

Python
515
star
53

surface-distance

Library to compute surface distance based performance metrics for segmentation tasks.
Python
509
star
54

tracr

Python
470
star
55

dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods
Jupyter Notebook
469
star
56

alphamissense

Python
461
star
57

long-form-factuality

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

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
438
star
59

lab2d

A customisable 2D platform for agent-based AI research
C++
417
star
60

open_x_embodiment

Jupyter Notebook
409
star
61

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.
Python
406
star
62

concordia

A library for generative social simulation
Python
385
star
63

clrs

Python
382
star
64

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment
Python
368
star
65

dm_pix

PIX is an image processing library in JAX, for JAX.
Python
366
star
66

mathematics_conjectures

Jupyter Notebook
348
star
67

alphastar

Python
346
star
68

dm_env

A Python interface for reinforcement learning environments
Python
333
star
69

spiral

We provide a pre-trained model for unconditional 19-step generation of CelebA-HQ images
C++
327
star
70

dm_robotics

Libraries, tools and tasks created and used at DeepMind Robotics.
Python
315
star
71

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).
Python
315
star
72

launchpad

Python
305
star
73

leo

Implementation of Meta-Learning with Latent Embedding Optimization
Python
302
star
74

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
C++
284
star
75

gqn-datasets

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

enn

Python
265
star
77

opro

official code for "Large Language Models as Optimizers"
Python
249
star
78

multi_object_datasets

Multi-object image datasets with ground-truth segmentation masks and generative factors.
Python
247
star
79

AQuA

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

card2code

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

arnheim

Jupyter Notebook
236
star
82

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
Python
236
star
83

synjax

Python
233
star
84

torch-hdf5

Torch interface to HDF5 library
Lua
232
star
85

dm_memorytasks

A set of 13 diverse machine-learning tasks that require memory to solve.
Python
220
star
86

Temporal-3D-Pose-Kinetics

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

dm_alchemy

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark
Python
197
star
88

neural_testbed

Jupyter Notebook
187
star
89

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Python
187
star
90

pg19

179
star
91

jmp

JMP is a Mixed Precision library for JAX.
Python
178
star
92

xquad

173
star
93

spectral_inference_networks

Implementation of Spectral Inference Networks, ICLR 2019
Python
165
star
94

abstract-reasoning-matrices

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

neural_networks_chomsky_hierarchy

Neural Networks and the Chomsky Hierarchy
Python
162
star
96

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.
C++
159
star
97

tensor_annotations

Annotating tensor shapes using Python types
Python
158
star
98

perception_test

Jupyter Notebook
155
star
99

mc_gradients

Jupyter Notebook
155
star
100

symplectic-gradient-adjustment

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