google-deepmind/mujoco

Multi-Joint dynamics with Contact. A general purpose physics simulator.

Language: Jupyter Notebook - Size: 104 MB - Last synced at: about 12 hours ago - Pushed at: 3 days ago - Stars: 9,378 - Forks: 972

google-deepmind/competitive_robot_table_tennis

Language: Jupyter Notebook - Size: 1.43 MB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 0 - Forks: 0

google-deepmind/dm_control

Google DeepMind's software stack for physics-based simulation and Reinforcement Learning environments, using MuJoCo.

Language: Python - Size: 94.9 MB - Last synced at: about 24 hours ago - Pushed at: 3 days ago - Stars: 4,047 - Forks: 693

google-deepmind/graphcast

Language: Python - Size: 1.52 MB - Last synced at: 1 day ago - Pushed at: 3 months ago - Stars: 6,002 - Forks: 749

google-deepmind/torax

TORAX: Tokamak transport simulation in JAX

Language: Python - Size: 325 MB - Last synced at: 1 day ago - Pushed at: 2 days ago - Stars: 418 - Forks: 50

google-deepmind/acme

A library of reinforcement learning components and agents

Language: Python - Size: 6.25 MB - Last synced at: 2 days ago - Pushed at: 9 days ago - Stars: 3,657 - Forks: 469

google-deepmind/concordia

A library for generative social simulation

Language: Python - Size: 3.46 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 855 - Forks: 186

google-deepmind/meltingpot

A suite of test scenarios for multi-agent reinforcement learning.

Language: Python - Size: 202 MB - Last synced at: 3 days ago - Pushed at: 24 days ago - Stars: 683 - Forks: 137

google-deepmind/open_spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

Language: C++ - Size: 29.4 MB - Last synced at: 3 days ago - Pushed at: 20 days ago - Stars: 4,490 - Forks: 981

google-deepmind/treescope

An interactive HTML pretty-printer for machine learning research in IPython notebooks.

Language: Python - Size: 840 KB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 407 - Forks: 23

google-deepmind/perception_test

Language: Jupyter Notebook - Size: 190 MB - Last synced at: 3 days ago - Pushed at: 4 days ago - Stars: 207 - Forks: 15

google-deepmind/alphafold3

AlphaFold 3 inference pipeline.

Language: Python - Size: 32.9 MB - Last synced at: 3 days ago - Pushed at: 13 days ago - Stars: 6,391 - Forks: 800

google-deepmind/gemma

Gemma open-weight LLM library, from Google DeepMind

Language: Jupyter Notebook - Size: 739 KB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 3,201 - Forks: 434

google-deepmind/graph_nets

Build Graph Nets in Tensorflow

Language: Python - Size: 1.43 MB - Last synced at: 1 day ago - Pushed at: over 2 years ago - Stars: 5,380 - Forks: 784

google-deepmind/latent-multi-hop-reasoning

[ACL 2024] Do Large Language Models Latently Perform Multi-Hop Reasoning?

Language: Python - Size: 9.51 MB - Last synced at: about 23 hours ago - Pushed at: about 1 month ago - Stars: 63 - Forks: 4

google-deepmind/mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control

Language: C++ - Size: 9.93 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 1,220 - Forks: 196

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

Language: Python - Size: 8.56 MB - Last synced at: 4 days ago - Pushed at: 5 days ago - Stars: 1,868 - Forks: 229

google-deepmind/disentangled_rnns

Language: Python - Size: 68.4 KB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 14 - Forks: 3

google-deepmind/alphafold

Open source code for AlphaFold 2.

Language: Python - Size: 15.9 MB - Last synced at: 5 days ago - Pushed at: about 1 month ago - Stars: 13,432 - Forks: 2,378

google-deepmind/xmanager

A platform for managing machine learning experiments

Language: Python - Size: 852 KB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 846 - Forks: 49

google-deepmind/alphatensor

Language: Python - Size: 7.31 MB - Last synced at: 4 days ago - Pushed at: about 1 year ago - Stars: 2,736 - Forks: 244

google-deepmind/deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications

Language: Jupyter Notebook - Size: 89.9 MB - Last synced at: 5 days ago - Pushed at: 9 days ago - Stars: 13,721 - Forks: 2,662

google-deepmind/kfac-jax

Second Order Optimization and Curvature Estimation with K-FAC in JAX.

Language: Python - Size: 839 KB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 268 - Forks: 26

google-deepmind/tapnet

Tracking Any Point (TAP)

Language: Jupyter Notebook - Size: 4.9 MB - Last synced at: 5 days ago - Pushed at: 6 days ago - Stars: 1,474 - Forks: 138

google-deepmind/physics-IQ-benchmark

Benchmarking physical understanding in generative video models

Language: Python - Size: 60.4 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 149 - Forks: 16

google-deepmind/loft

LOFT: A 1 Million+ Token Long-Context Benchmark

Language: Python - Size: 323 KB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 187 - Forks: 14

google-deepmind/jraph

A Graph Neural Network Library in Jax

Language: Python - Size: 237 KB - Last synced at: 6 days ago - Pushed at: about 1 year ago - Stars: 1,424 - Forks: 95

google-deepmind/sonnet

TensorFlow-based neural network library

Language: Python - Size: 16.1 MB - Last synced at: 4 days ago - Pushed at: 2 months ago - Stars: 9,842 - Forks: 1,302

google-deepmind/simply

Language: Python - Size: 0 Bytes - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 2 - Forks: 1

google-deepmind/mujoco_menagerie

A collection of high-quality models for the MuJoCo physics engine, curated by Google DeepMind.

Language: Python - Size: 341 MB - Last synced at: 6 days ago - Pushed at: 6 days ago - Stars: 1,968 - Forks: 285

google-deepmind/dm_robotics

Libraries, tools and tasks created and used at DeepMind Robotics.

Language: Python - Size: 91.7 MB - Last synced at: 6 days ago - Pushed at: 5 months ago - Stars: 367 - Forks: 42

google-deepmind/android_env

RL research on Android devices.

Language: Python - Size: 186 MB - Last synced at: 5 days ago - Pushed at: 16 days ago - Stars: 1,072 - Forks: 83

google-deepmind/dangerous-capability-evaluations

Language: Python - Size: 74.2 KB - Last synced at: 2 days ago - Pushed at: 7 months ago - Stars: 54 - Forks: 3

google-deepmind/penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

Language: Python - Size: 484 MB - Last synced at: 3 days ago - Pushed at: 19 days ago - Stars: 1,758 - Forks: 62

google-deepmind/tensor_annotations

Annotating tensor shapes using Python types

Language: Python - Size: 386 KB - Last synced at: 2 days ago - Pushed at: almost 2 years ago - Stars: 159 - Forks: 13

google-deepmind/mujoco_warp

GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardware.

Language: Python - Size: 1.13 MB - Last synced at: 8 days ago - Pushed at: 9 days ago - Stars: 369 - Forks: 34

google-deepmind/dnc

A TensorFlow implementation of the Differentiable Neural Computer.

Language: Python - Size: 71.3 KB - Last synced at: 1 day ago - Pushed at: almost 4 years ago - Stars: 2,513 - Forks: 441

google-deepmind/dm-haiku

JAX-based neural network library

Language: Python - Size: 2.49 MB - Last synced at: 9 days ago - Pushed at: 17 days ago - Stars: 3,012 - Forks: 243

google-deepmind/reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research

Language: C++ - Size: 1.87 MB - Last synced at: 2 days ago - Pushed at: about 2 months ago - Stars: 724 - Forks: 100

google-deepmind/trecvit

Language: Python - Size: 492 KB - Last synced at: 9 days ago - Pushed at: 9 days ago - Stars: 1 - Forks: 0

google-deepmind/neural-processes

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

Language: Jupyter Notebook - Size: 1.21 MB - Last synced at: 9 days ago - Pushed at: over 4 years ago - Stars: 1,000 - Forks: 151

google-deepmind/pysc2

StarCraft II Learning Environment

Language: Python - Size: 4.36 MB - Last synced at: 10 days ago - Pushed at: 9 months ago - Stars: 8,109 - Forks: 1,160

google-deepmind/educational

Language: Jupyter Notebook - Size: 430 KB - Last synced at: 7 days ago - Pushed at: over 2 years ago - Stars: 1,429 - Forks: 195

google-deepmind/actionengine-ts

Language: TypeScript - Size: 452 KB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 1 - Forks: 0

google-deepmind/personality_in_llms

Language: Jupyter Notebook - Size: 417 KB - Last synced at: 11 days ago - Pushed at: 11 days ago - Stars: 6 - Forks: 2

google-deepmind/opro

official code for "Large Language Models as Optimizers"

Language: Python - Size: 4.51 MB - Last synced at: 11 days ago - Pushed at: 5 months ago - Stars: 536 - Forks: 60

google-deepmind/boxoban-levels

This repository contains levels for boxoban, a box-pushing puzzle game inspired by Sokoban.

Size: 31.5 MB - Last synced at: about 19 hours ago - Pushed at: over 2 years ago - Stars: 70 - Forks: 20

google-deepmind/einshape

Language: Python - Size: 67.4 KB - Last synced at: about 24 hours ago - Pushed at: 10 months ago - Stars: 102 - Forks: 8

google-deepmind/chex

Language: Python - Size: 578 KB - Last synced at: 2 days ago - Pushed at: 24 days ago - Stars: 845 - Forks: 49

google-deepmind/mujoco_playground

An open-source library for GPU-accelerated robot learning and sim-to-real transfer.

Language: Jupyter Notebook - Size: 283 MB - Last synced at: 14 days ago - Pushed at: 15 days ago - Stars: 916 - Forks: 108

google-deepmind/dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods

Language: Jupyter Notebook - Size: 28.2 MB - Last synced at: 11 days ago - Pushed at: over 4 years ago - Stars: 499 - Forks: 68

google-deepmind/neural_testbed

Language: Jupyter Notebook - Size: 2.97 MB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 192 - Forks: 18

google-deepmind/learning-to-learn

Learning to Learn in TensorFlow

Language: Python - Size: 87.9 KB - Last synced at: 13 days ago - Pushed at: almost 4 years ago - Stars: 4,060 - Forks: 603

google-deepmind/mathematics_dataset

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

Language: Python - Size: 60.5 KB - Last synced at: 15 days ago - Pushed at: 4 months ago - Stars: 1,868 - Forks: 255

google-deepmind/mctx

Monte Carlo tree search in JAX

Language: Python - Size: 476 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 2,454 - Forks: 199

google-deepmind/tree

tree is a library for working with nested data structures

Language: Python - Size: 152 KB - Last synced at: 13 days ago - Pushed at: 3 months ago - Stars: 977 - Forks: 61

google-deepmind/bbeh

Language: Python - Size: 2.36 MB - Last synced at: 15 days ago - Pushed at: about 1 month ago - Stars: 64 - Forks: 2

google-deepmind/hanabi-learning-environment 📦

hanabi_learning_environment is a research platform for Hanabi experiments.

Language: Python - Size: 137 KB - Last synced at: 12 days ago - Pushed at: about 2 years ago - Stars: 650 - Forks: 156

google-deepmind/tips

Language: Jupyter Notebook - Size: 5.09 MB - Last synced at: 18 days ago - Pushed at: 18 days ago - Stars: 31 - Forks: 3

google-deepmind/distrax

Language: Python - Size: 624 KB - Last synced at: 16 days ago - Pushed at: 16 days ago - Stars: 564 - Forks: 32

google-deepmind/rc-data 📦

Question answering dataset featured in "Teaching Machines to Read and Comprehend

Language: Python - Size: 1.26 MB - Last synced at: 12 days ago - Pushed at: about 8 years ago - Stars: 1,294 - Forks: 243

google-deepmind/alphageometry

Language: Python - Size: 176 KB - Last synced at: 17 days ago - Pushed at: 6 months ago - Stars: 4,439 - Forks: 505

google-deepmind/rlax

Language: Python - Size: 488 KB - Last synced at: 14 days ago - Pushed at: 16 days ago - Stars: 1,305 - Forks: 91

google-deepmind/launchpad 📦

Language: Python - Size: 672 KB - Last synced at: 18 days ago - Pushed at: over 1 year ago - Stars: 320 - Forks: 41

google-deepmind/lab

A customisable 3D platform for agent-based AI research

Language: C - Size: 439 MB - Last synced at: 16 days ago - Pushed at: over 2 years ago - Stars: 7,207 - Forks: 1,374

google-deepmind/dks

Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural network models (and their initializations) to make them easier to train.

Language: Python - Size: 1.23 MB - Last synced at: 6 days ago - Pushed at: 9 days ago - Stars: 70 - Forks: 5

google-deepmind/tf2jax

Language: Python - Size: 414 KB - Last synced at: 15 days ago - Pushed at: 17 days ago - Stars: 114 - Forks: 12

google-deepmind/scalable_agent

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

Language: Python - Size: 52.7 KB - Last synced at: 12 days ago - Pushed at: about 6 years ago - Stars: 1,002 - Forks: 161

google-deepmind/kinetics-i3d

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

Language: Python - Size: 388 MB - Last synced at: 19 days ago - Pushed at: over 5 years ago - Stars: 1,784 - Forks: 471

google-deepmind/fusion_surrogates

A library of surrogate transport models for tokamak fusion.

Language: Python - Size: 304 KB - Last synced at: 19 days ago - Pushed at: 19 days ago - Stars: 4 - Forks: 1

google-deepmind/surface-distance

Library to compute surface distance based performance metrics for segmentation tasks.

Language: Python - Size: 39.1 KB - Last synced at: 18 days ago - Pushed at: 2 months ago - Stars: 561 - Forks: 117

google-deepmind/trfl

TensorFlow Reinforcement Learning

Language: Python - Size: 284 KB - Last synced at: 13 days ago - Pushed at: over 2 years ago - Stars: 3,138 - Forks: 387

google-deepmind/tracr

Language: Python - Size: 146 KB - Last synced at: 21 days ago - Pushed at: about 1 year ago - Stars: 528 - Forks: 45

google-deepmind/levintreesearch_cm

Language: Racket - Size: 16.1 MB - Last synced at: about 19 hours ago - Pushed at: 2 months ago - Stars: 29 - Forks: 5

google-deepmind/envlogger

A tool for recording RL trajectories.

Language: Python - Size: 444 KB - Last synced at: 12 days ago - Pushed at: 6 months ago - Stars: 101 - Forks: 14

google-deepmind/multidim-image-augmentation

This package provides TensorFlow Ops for multidimensional volumetric image augmentation.

Language: C++ - Size: 3.17 MB - Last synced at: 6 days ago - Pushed at: 3 months ago - Stars: 135 - Forks: 34

google-deepmind/PGMax

Loopy belief propagation for factor graphs on discrete variables in JAX

Language: Jupyter Notebook - Size: 14.6 MB - Last synced at: 23 days ago - Pushed at: 6 months ago - Stars: 145 - Forks: 11

google-deepmind/jmp

JMP is a Mixed Precision library for JAX.

Language: Python - Size: 41 KB - Last synced at: 15 days ago - Pushed at: 3 months ago - Stars: 194 - Forks: 16

google-deepmind/questbench

Language: Python - Size: 538 KB - Last synced at: 24 days ago - Pushed at: 24 days ago - Stars: 4 - Forks: 0

google-deepmind/pycolab

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

Language: Python - Size: 438 KB - Last synced at: 8 days ago - Pushed at: over 5 years ago - Stars: 659 - Forks: 123

google-deepmind/mishax

Language: Python - Size: 58.6 KB - Last synced at: 26 days ago - Pushed at: 26 days ago - Stars: 124 - Forks: 8

google-deepmind/enn

Language: Python - Size: 1.41 MB - Last synced at: 28 days ago - Pushed at: about 2 months ago - Stars: 309 - Forks: 61

google-deepmind/codoc 📦

Language: Python - Size: 337 KB - Last synced at: 22 days ago - Pushed at: almost 2 years ago - Stars: 115 - Forks: 21

google-deepmind/bsuite 📦

bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent

Language: Python - Size: 1.96 MB - Last synced at: 2 days ago - Pushed at: about 1 year ago - Stars: 1,518 - Forks: 186

google-deepmind/open_x_embodiment

Language: Jupyter Notebook - Size: 19.7 MB - Last synced at: about 1 month ago - Pushed at: 5 months ago - Stars: 1,113 - Forks: 72

google-deepmind/funsearch

Language: Jupyter Notebook - Size: 1.03 MB - Last synced at: about 1 month ago - Pushed at: about 1 year ago - Stars: 800 - Forks: 148

google-deepmind/ai-safety-gridworlds 📦

This is a suite of reinforcement learning environments illustrating various safety properties of intelligent agents.

Language: Python - Size: 107 KB - Last synced at: 12 days ago - Pushed at: almost 3 years ago - Stars: 615 - Forks: 125

google-deepmind/jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX

Language: Python - Size: 197 KB - Last synced at: 15 days ago - Pushed at: 10 months ago - Stars: 93 - Forks: 12

google-deepmind/alphatensor_quantum

Language: Python - Size: 521 KB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 13 - Forks: 5

google-deepmind/mc_gradients

Language: Jupyter Notebook - Size: 298 KB - Last synced at: about 1 month ago - Pushed at: 9 months ago - Stars: 166 - Forks: 39

google-deepmind/dm_nevis

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

Language: Python - Size: 390 KB - Last synced at: about 1 month ago - Pushed at: over 2 years ago - Stars: 101 - Forks: 6

google-deepmind/AQuA 📦

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

Size: 29.6 MB - Last synced at: about 1 month ago - Pushed at: over 7 years ago - Stars: 311 - Forks: 47

google-deepmind/recoglab

Language: Python - Size: 103 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 1 - Forks: 0

google-deepmind/long-form-factuality

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

Language: Python - Size: 759 KB - Last synced at: about 1 month ago - Pushed at: about 2 months ago - Stars: 589 - Forks: 71

google-deepmind/pg19

Size: 759 KB - Last synced at: 23 days ago - Pushed at: about 5 years ago - Stars: 239 - Forks: 19

google-deepmind/simulation_streams

Simulation Streams is a programming paradigm designed to efficiently control and leverage Large Language Models (LLMs) for complex, dynamic simulations and agentic workflows.

Language: Python - Size: 93.8 KB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 14 - Forks: 2

google-deepmind/nuclease_design

ML-guided enzyme engineering

Language: Jupyter Notebook - Size: 37.9 MB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 41 - Forks: 12

google-deepmind/jaxline

Language: Python - Size: 62.5 KB - Last synced at: 17 days ago - Pushed at: over 1 year ago - Stars: 154 - Forks: 22

google-deepmind/materials_discovery

Language: Jupyter Notebook - Size: 32.4 MB - Last synced at: about 2 months ago - Pushed at: about 2 months ago - Stars: 951 - Forks: 153

google-deepmind/3d-shapes

This repository contains the 3D shapes dataset, used in Kim, Hyunjik and Mnih, Andriy. "Disentangling by Factorising." In Proceedings of the 35th International Conference on Machine Learning (ICML). 2018. to assess the disentanglement properties of unsupervised learning methods.

Language: Jupyter Notebook - Size: 579 KB - Last synced at: about 1 month ago - Pushed at: about 1 year ago - Stars: 140 - Forks: 33