Google DeepMind

Official repository for "VideoPrism: A Foundational Visual Encoder for Video Understanding" (ICML 2024)

A tool for recording RL trajectories.

GeeFlow - generate and process large-scale geospatial datasets with Google Earth Engine.

Library for sequence-to-sequence numeric prediction, applicable to any tokenizable input, and allows pretraining and fine-tuning over multiple tasks.

A networking protocol for agent-environment communication

Benchmarking physical understanding in generative video models

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

official code for "Large Language Models as Optimizers"

Minimal and scalable research codebase in JAX, designed for rapid iteration on frontier research in LLM and other autoregressive models.

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

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

RL research on Android devices.

Jeo: Jax model training lib for Earth Observation

A customisable 2D platform for agent-based AI research

Neural Networks and the Chomsky Hierarchy

A gradient processing and optimization library in JAX.

A Jax package for approximate curvature estimation and optimization using KFAC.

Restoring and attributing ancient texts using deep neural networks

Datasets to protect Earth's forests and biodiversity

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

NEVIS'22: Benchmarking the next generation of never-ending learners

Accompanying code for "Discovering State-of-the-art Reinforcement Algorithms" Nature publication

TensorFlow-based neural network library

A Python SDK for interacting and visualizing genomic models.

Tracking Any Point (TAP)

Monte Carlo tree search in JAX.

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

JAX-based neural network library

A library for generative social simulation

GDM Science Skills to speed up agentic scientific workflows with better grounding and higher token efficiency. Integrate insights from AlphaGenome, AFDB, UniProt and 30+ other databases and tools.

AlphaFold 3 inference pipeline.

Gemma open-weight LLM library from Google DeepMind.

A customisable 3D platform for agent-based AI research

Build Graph Nets in Tensorflow