• Stars
    star
    1,398
  • Rank 33,617 (Top 0.7 %)
  • Language
    Jupyter Notebook
  • License
    Apache License 2.0
  • Created about 4 years ago
  • Updated about 2 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

DeepMind Educational Resources

This repository contains a collection of educational tutorials that we have prepared for teaching the basics of machine learning to various audiences. The goal is to have simple and accessible resources that can enable anyone, including those with no machine learning background, to engage and learn from these tutorials.

Our aim is to contribute to democratisation of machine learning by providing accessible educational resources to inspire everyone.

The tutorials are presented as notebooks that can be launched via Google Colab.

Introductory Tutorials

Fluttering Avians Open In Colab

Videogames and artificial intelligence have a long and happy history together. In this tutorial we will play a familiar (and hard!) game, learn what an agent is, and how we can make them learn to play the game with superhuman abilities. The agents we will build are evaluated on their performance on the game, and selected with variation in a virtual circle of life inspired by evolution.

Fun with Language Open In Colab

This tutorial shows how we can build important artificial intelligence models for natural language processing called language models. Language models are surprisingly effective models that learn to predict the next letter (or word) given previous letters (or words) --- they essentially learn which letters (or words) go well together. In this tutorial we teach you how the computer represents and processes language, and show how we can use a big chunk of text to learn language models and apply them on two tasks: decoding secret messages and generating text.

Generative Models Open In Colab

Creativity is central to human intelligence. In this tutorial we see where Artificial Intelligence (AI) meets creativity. We will show you how an AI can produce realistic looking images of everyday objects as well as works-of-art and imagine "spider-dogs". We refer to an AI that is able to do any one of these things as Generative Models and towards the end of this tutorial you will build your own one.

Protein Folding Open In Colab

This tutorial explores the use of machine learning for solving the protein structure prediction problem. Although the tutorial does not actually solve the problem itself, it provides students with very basic background in coding and biology to get started, and trains their intuition on machine learning methods, with the help of visualisation and a few examples of folding simple protein structures.

Basics of Reinforcement Learning Open In Colab

This tutorial introduces students to a simple reinforcement learning (RL) setup used in research. It involves running pre-existing code to set up an RL environment and visualise it. Students can then look at how a completely random (untrained) agent behaves in these environments. We also include simple code that implements a reinforcement learning method that can train the agent to solve these simple tasks. The behaviour of the trained agent can be then visualised together with plots of how the agent evolves through training.

Scientific Thinking Open In Colab

This tutorial teaches you the basic ideas that underline scientific thinking. We cover developing and testing new knowledge with the scientific method through experimentation and validation, while showing common pitfalls in the process. Through a series of games you will play as an agent trying to understand the world, you will get insight into some of the core ideas behind scientific thinking.

Summer Schools Tutorials

The following tutorials are intended to be used by anyone who wishes to teach the basics of machine learning and artificial intelligence at a summer school, introductory university course or community meetup. The initial versions of these Colabs were written by a group of volunteers inside DeepMind, drawn on our experience in teaching at:

  • Deep Learning Indaba
  • Eastern European Machine Learning Summer School
  • Khipu: Latin American Meeting In Artificial Intelligence
  • Mediterranean Machine Learning Summer School
  • Southeast Asia Machine Learning School

The material as a whole does not form a structured course. However, each Colab or group of Colabs tell their own story, so that they can also be used as self-learning materials by students who don’t have direct access to a teacher. They are released under an Apache license, which means that you could adapt them to your own unique style of teaching. You are welcome to adapt them in any way you like when you create teaching material for your own community meetup or summer school.

The Colabs are written in Python and Jax, which are some of the tools we use at DeepMind. The current Colabs in this repository are:

Introduction to Python and NumPy for Deep Learning Open In Colab

This tutorial lays the basics of Python and NumPy often used in the field of Deep Learning. The tutorial presents the fundamentals of Python from scratch, together with all the standard programming constructs and all frequently used libraries and packages (such as functools, type annotations, debugging, and basic plotting). The tutorial introduces common NumPy functions that underpin most of the tensor transformations used in both Deep Learning and wider Machine Learning. It is intended as a reference for Python and NumPy to be used across colabs in this repository, particularly for the Deep Learning / Machine Learning colabs. However, please note that this tutorial is not intended to be an exhaustive resource for Python and NumPy.

Introduction to Supervised Learning 1 - Regression Open In Colab

This tutorial presents a gentle and intuitive introduction to supervised learning. The tutorial is intended to be taught from the front of a class to a group of students, but could also be done individually. The student learns what a model for data is and what model parameters are, and experiences basic optimization of a loss function to fit data first-hand. We introduce automatic differentiation and Jax. The tutorial concludes with first steps toward non-linear models, underfitting, overfitting and basic regularization.

Introduction to Unsupervised Learning Open In Colab

This tutorial covers the basics of unsupervised learning. In the real world, this is the most common form of learning: where no labels or feedback are available. How then is the machine supposed to learn? The most important cost function guiding the machine is negative log likelihood (aka log-loss). We explain what it means in a gentle introduction to the probability theory (density estimation, maximum likelihood/log-loss and latent variable modelling). We show you how to build and train an autoencoder, and how to compress an image using the K−means clustering algorithm - all using only basic Python commands! Lots of visualisations and exercises are included to make this journey fun.

Introduction to Reinforcement Learning Open In Colab

This tutorial presents the basics of Reinforcement Learning - the basics of training an agent which acts in an environment to maximise a numerical reward it gets from the said environment for its actions. In this tutorial, we first showcase how to define a simple Catch environment and a random agent operating in it. Following this, we present three increasingly complex (and thus capable) agents. First, we implement a Q-learning agent as one of the simplest examples of value-based reinforcement learning agents which relies on tracking the quality of state-action combinations (Q values) with a simple look-up table. Second, we upgrade the Q-learning agent to use a neural network for tracking the Q values. Finally, we upgrade the neural net-based Q-learning agent to a full Deep Q-Learning Agent (DQN), which further improves the neural-network-based estimation of the Q values.

Introduction to Graph Neural Nets with JAX/jraph Open In Colab

This tutorial teaches you the fundamentals of graph neural networks (GNNs), a family of architectures designed for learning on graph-structured data consisting of nodes and edges, where the edges describe relationships between the nodes. E.g. a molecule can be represented as a graph with atoms as nodes, and the chemical bonds between them as edges. Predicting chemical properties of molecules is an example applications of GNNs, along with many others, e.g. predicting relationships between papers in a citation network.

Differently from previous setups where we have seen input features in the form of a real-valued vectors (e.g. vectorised form of images), the structure of the data itself is important here, e.g. how to atoms are connected to each other in a molecule may influence its properties. GNNs allow us to take this graph structure explicitly into account during training and inference, which is something other architectures like fully-connected networks or recurrent neural networks would not be able to do.

We will go over the basics of graph theory, code GNNs from scratch, and apply our models on three types of graph learning tasks. More specifically, we will implement two common types of GNN architectures: Graph Convolutional Networks and Graph Attention Networks. We will also introduce you to jraph, which is a library designed for working with graph neural networks in JAX.

Call for contributions

This repository is an ongoing effort, maintained by volunteers. As you’ve read to this point, we now count you as a co-volunteer! You are welcome to help improve the material through pull requests, bug reports, features requests and any other code contributions!

Contact

If you have any feedback, or would like to get in touch with us, please reach out by opening a new issue on the GitHub repo or emailing us at [email protected].

Disclaimer

This is not an officially supported Google product.

More Repositories

1

deepmind-research

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

alphafold

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

sonnet

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

mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.
Jupyter Notebook
8,113
star
5

pysc2

StarCraft II Learning Environment
Python
8,001
star
6

lab

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

graph_nets

Build Graph Nets in Tensorflow
Python
5,352
star
8

graphcast

Python
4,517
star
9

open_spiel

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

alphageometry

Python
4,079
star
11

learning-to-learn

Learning to Learn in TensorFlow
Python
4,064
star
12

dm_control

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

acme

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

trfl

TensorFlow Reinforcement Learning
Python
3,136
star
15

dm-haiku

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

alphatensor

Python
2,670
star
17

dnc

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

gemma

Open weights LLM from Google DeepMind.
Python
2,421
star
19

mctx

Monte Carlo tree search in JAX
Python
2,313
star
20

code_contests

C++
2,064
star
21

optax

Optax is a gradient processing and optimization library for JAX.
Python
1,670
star
22

kinetics-i3d

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

penzai

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

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
25

bsuite

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

jraph

A Graph Neural Network Library in Jax
Python
1,349
star
27

rc-data

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

mujoco_menagerie

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

tapnet

Tracking Any Point (TAP)
Jupyter Notebook
1,266
star
30

rlax

Python
1,223
star
31

scalable_agent

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

android_env

RL research on Android devices.
Python
977
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
969
star
34

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control
C++
959
star
35

dramatron

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

tree

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

materials_discovery

Jupyter Notebook
866
star
38

xmanager

A platform for managing machine learning experiments
Python
815
star
39

open_x_embodiment

Jupyter Notebook
785
star
40

chex

Python
751
star
41

ferminet

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

reverb

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

funsearch

Jupyter Notebook
699
star
44

alphadev

Python
688
star
45

pycolab

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

concordia

A library for generative social simulation
Python
634
star
47

hanabi-learning-environment

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

recurrentgemma

Open weights language model from Google DeepMind, based on Griffin.
Python
603
star
49

ai-safety-gridworlds

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

meltingpot

A suite of test scenarios for multi-agent reinforcement learning.
Python
576
star
51

ithaca

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

dqn

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

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
534
star
54

distrax

Python
527
star
55

long-form-factuality

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

surface-distance

Library to compute surface distance based performance metrics for segmentation tasks.
Python
526
star
57

tracr

Python
496
star
58

alphamissense

Python
494
star
59

dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods
Jupyter Notebook
476
star
60

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
452
star
61

clrs

Jupyter Notebook
444
star
62

lab2d

A customisable 2D platform for agent-based AI research
C++
420
star
63

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
64

alphastar

Python
403
star
65

dm_pix

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

opro

official code for "Large Language Models as Optimizers"
Python
383
star
67

mathematics_conjectures

Jupyter Notebook
367
star
68

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment
Python
367
star
69

torax

TORAX: Tokamak transport simulation in JAX
Python
361
star
70

dm_env

A Python interface for reinforcement learning environments
Python
343
star
71

dm_robotics

Libraries, tools and tasks created and used at DeepMind Robotics.
Python
341
star
72

spiral

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

launchpad

Python
310
star
74

leo

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

enn

Python
291
star
76

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++
285
star
77

gqn-datasets

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

treescope

An interactive HTML pretty-printer for machine learning research in IPython notebooks.
Python
256
star
79

multi_object_datasets

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

AQuA

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

synjax

Python
238
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

card2code

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

arnheim

Jupyter Notebook
235
star
85

torch-hdf5

Torch interface to HDF5 library
Lua
234
star
86

kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.
Python
231
star
87

dm_memorytasks

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

Temporal-3D-Pose-Kinetics

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

dm_alchemy

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

neural_testbed

Jupyter Notebook
191
star
91

perception_test

Jupyter Notebook
184
star
92

jmp

JMP is a Mixed Precision library for JAX.
Python
183
star
93

neural_networks_chomsky_hierarchy

Neural Networks and the Chomsky Hierarchy
Python
183
star
94

xquad

180
star
95

nanodo

Python
180
star
96

pg19

179
star
97

spectral_inference_networks

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

barkour_robot

Barkour Robot: Agile Quadruped Robots by Google DeepMind
C++
165
star
99

onetwo

Python
164
star
100

abstract-reasoning-matrices

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