google-deepmind/mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control

Language: C++ - Size: 9.98 MB - Last synced at: about 8 hours ago - Pushed at: 11 days ago - Stars: 1,404 - Forks: 226

google-deepmind/mujoco_playground

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

Language: Jupyter Notebook - Size: 322 MB - Last synced at: about 24 hours ago - Pushed at: 1 day ago - Stars: 1,378 - Forks: 194

google-deepmind/mujoco

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

Language: C++ - Size: 110 MB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 10,531 - Forks: 1,105

google-deepmind/sonnet

TensorFlow-based neural network library

Language: Python - Size: 16 MB - Last synced at: 1 day ago - Pushed at: about 1 month ago - Stars: 9,879 - Forks: 1,307

google-deepmind/meltingpot

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

Language: Python - Size: 202 MB - Last synced at: about 21 hours ago - Pushed at: 4 days ago - Stars: 732 - Forks: 140

google-deepmind/jax_privacy

Algorithms for Privacy-Preserving Machine Learning in JAX

Language: Python - Size: 754 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 96 - Forks: 14

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,387 - Forks: 786

google-deepmind/xarray_jax

Language: Python - Size: 56.6 KB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 10 - Forks: 2

google-deepmind/concordia

A library for generative social simulation

Language: Python - Size: 3.6 MB - Last synced at: about 20 hours ago - Pushed at: 2 days ago - Stars: 1,014 - Forks: 214

google-deepmind/torax

TORAX: Tokamak transport simulation in JAX

Language: Python - Size: 569 MB - Last synced at: 2 days ago - Pushed at: 2 days ago - Stars: 467 - Forks: 65

google-deepmind/formal-conjectures

A collection of formalized statements of conjectures in Lean.

Language: Lean - Size: 1.06 MB - Last synced at: 3 days ago - Pushed at: 3 days ago - Stars: 610 - Forks: 76

google-deepmind/android_env

RL research on Android devices.

Language: Python - Size: 186 MB - Last synced at: 3 days ago - Pushed at: 22 days ago - Stars: 1,128 - Forks: 95

google-deepmind/dm_robotics

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

Language: Python - Size: 91.7 MB - Last synced at: 1 day ago - Pushed at: about 1 month ago - Stars: 384 - Forks: 42

google-deepmind/dm-haiku

JAX-based neural network library

Language: Python - Size: 2.45 MB - Last synced at: 3 days ago - Pushed at: 4 days ago - Stars: 3,081 - Forks: 253

google-deepmind/alphagenome

This API provides programmatic access to the AlphaGenome model developed by Google DeepMind.

Language: Python - Size: 5.77 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 1,184 - Forks: 136

google-deepmind/optax

Optax is a gradient processing and optimization library for JAX.

Language: Python - Size: 8 MB - Last synced at: 3 days ago - Pushed at: 10 days ago - Stars: 1,996 - Forks: 255

google-deepmind/mctx

Monte Carlo tree search in JAX

Language: Python - Size: 480 KB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 2,526 - Forks: 208

google-deepmind/rlax

Language: Python - Size: 504 KB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 1,355 - Forks: 92

google-deepmind/limit

On the Theoretical Limitations of Embedding-Based Retrieval

Language: Jupyter Notebook - Size: 6.44 MB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 380 - Forks: 22

google-deepmind/chex

Language: Python - Size: 506 KB - Last synced at: 4 days ago - Pushed at: 4 days ago - Stars: 880 - Forks: 56

google-deepmind/mujoco_warp

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

Language: Python - Size: 5.47 MB - Last synced at: 5 days ago - Pushed at: 5 days ago - Stars: 703 - Forks: 70

google-deepmind/graphcast

Language: Python - Size: 1.51 MB - Last synced at: 5 days ago - Pushed at: 2 months ago - Stars: 6,279 - Forks: 795

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: 5 days ago - Pushed at: 6 months ago - Stars: 76 - Forks: 6

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: 3 days ago - Pushed at: 9 months ago - Stars: 1,897 - Forks: 258

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: 6 days ago - Pushed at: 8 days ago - Stars: 4,214 - Forks: 709

google-deepmind/regress-lm

Library for text-to-text regression, applicable to any input string representation and allows pretraining and fine-tuning over multiple regression tasks.

Language: Python - Size: 119 KB - Last synced at: 7 days ago - Pushed at: 7 days ago - Stars: 184 - Forks: 27

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: 7 days ago - Pushed at: over 1 year ago - Stars: 1,526 - Forks: 188

google-deepmind/mujoco_menagerie

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

Language: Python - Size: 404 MB - Last synced at: 7 days ago - Pushed at: 29 days ago - Stars: 2,414 - Forks: 339

google-deepmind/kfac-jax

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

Language: Python - Size: 907 KB - Last synced at: 6 days ago - Pushed at: about 2 months ago - Stars: 281 - Forks: 28

google-deepmind/alphafold

Open source code for AlphaFold 2.

Language: Python - Size: 16 MB - Last synced at: 7 days ago - Pushed at: 15 days ago - Stars: 13,786 - Forks: 2,470

google-deepmind/acme

A library of reinforcement learning components and agents

Language: Python - Size: 6.3 MB - Last synced at: 7 days ago - Pushed at: about 2 months ago - Stars: 3,777 - Forks: 497

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: 7 days ago - Pushed at: 14 days ago - Stars: 14,246 - Forks: 2,740

google-deepmind/dramatron

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

Language: Jupyter Notebook - Size: 1.32 MB - Last synced at: 3 days ago - Pushed at: about 1 year ago - Stars: 1,013 - Forks: 172

google-deepmind/lab

A customisable 3D platform for agent-based AI research

Language: C - Size: 439 MB - Last synced at: 6 days ago - Pushed at: over 2 years ago - Stars: 7,259 - Forks: 1,387

google-deepmind/synthid-text

Language: Python - Size: 2.22 MB - Last synced at: 8 days ago - Pushed at: about 1 month ago - Stars: 542 - Forks: 55

google-deepmind/videoprism

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

Language: Python - Size: 1.47 MB - Last synced at: 8 days ago - Pushed at: 8 days ago - Stars: 281 - Forks: 24

google-deepmind/ai-foundations

Language: Jupyter Notebook - Size: 144 KB - Last synced at: 8 days ago - Pushed at: 9 days ago - Stars: 12 - Forks: 4

google-deepmind/alphatensor

Language: Python - Size: 7.31 MB - Last synced at: 8 days ago - Pushed at: over 1 year ago - Stars: 2,780 - Forks: 254

google-deepmind/code_contests 📦

Language: C++ - Size: 71.3 KB - Last synced at: 9 days ago - Pushed at: almost 2 years ago - Stars: 2,159 - Forks: 219

google-deepmind/tensor_annotations

Annotating tensor shapes using Python types

Language: Python - Size: 386 KB - Last synced at: 7 days ago - Pushed at: about 2 years ago - Stars: 158 - Forks: 13

google-deepmind/jraph 📦

A Graph Neural Network Library in Jax

Language: Python - Size: 237 KB - Last synced at: 5 days ago - Pushed at: over 1 year ago - Stars: 1,443 - Forks: 98

google-deepmind/meqpy

Language: Python - Size: 119 KB - Last synced at: 9 days ago - Pushed at: 10 days ago - Stars: 2 - Forks: 0

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.9 MB - Last synced at: 9 days ago - Pushed at: 27 days ago - Stars: 741 - Forks: 102

google-deepmind/distrax

Language: Python - Size: 638 KB - Last synced at: 9 days ago - Pushed at: 10 days ago - Stars: 588 - Forks: 38

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.8 MB - Last synced at: 9 days ago - Pushed at: 10 days ago - Stars: 4,703 - Forks: 1,018

google-deepmind/gemma

Gemma open-weight LLM library, from Google DeepMind

Language: Python - Size: 1000 KB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 3,647 - Forks: 526

google-deepmind/disentangled_rnns

Language: Python - Size: 197 KB - Last synced at: 10 days ago - Pushed at: 10 days ago - Stars: 26 - Forks: 5

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: 6 days ago - Pushed at: over 4 years ago - Stars: 1,003 - Forks: 153

google-deepmind/librispeech-long

LibriSpeech-Long is a benchmark dataset for long-form speech generation and processing. Released as part of "Long-Form Speech Generation with Spoken Language Models" (arXiv 2024).

Size: 9.77 KB - Last synced at: 5 days ago - Pushed at: 8 months ago - Stars: 84 - Forks: 3

google-deepmind/dangerous-capability-evaluations

Language: Python - Size: 220 KB - Last synced at: 7 days ago - Pushed at: about 1 month ago - Stars: 57 - Forks: 4

google-deepmind/xmanager

A platform for managing machine learning experiments

Language: Python - Size: 912 KB - Last synced at: 11 days ago - Pushed at: 12 days ago - Stars: 867 - Forks: 52

google-deepmind/gemini-robotics-sdk

Language: Python - Size: 253 KB - Last synced at: 11 days ago - Pushed at: 2 months ago - Stars: 449 - Forks: 27

google-deepmind/penzai

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

Language: Python - Size: 484 MB - Last synced at: 8 days ago - Pushed at: 3 months ago - Stars: 1,813 - Forks: 65

google-deepmind/learning-to-learn

Learning to Learn in TensorFlow

Language: Python - Size: 87.9 KB - Last synced at: 7 days ago - Pushed at: about 4 years ago - Stars: 4,062 - Forks: 606

google-deepmind/pysc2

StarCraft II Learning Environment

Language: Python - Size: 4.36 MB - Last synced at: 12 days ago - Pushed at: about 1 year ago - Stars: 8,170 - Forks: 1,169

google-deepmind/scivid

Language: Python - Size: 2.96 MB - Last synced at: 15 days ago - Pushed at: 15 days ago - Stars: 12 - Forks: 0

google-deepmind/treescope

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

Language: Python - Size: 862 KB - Last synced at: 8 days ago - Pushed at: 29 days ago - Stars: 443 - Forks: 24

google-deepmind/ferminet

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

Language: Python - Size: 488 KB - Last synced at: 12 days ago - Pushed at: 3 months ago - Stars: 780 - Forks: 153

google-deepmind/jmp

JMP is a Mixed Precision library for JAX.

Language: Python - Size: 41 KB - Last synced at: 13 days ago - Pushed at: 7 months ago - Stars: 208 - Forks: 17

google-deepmind/bagz

Bagz is a format for storing a sequence of string records. It supports per-record compression and fast index-based lookup.

Language: C++ - Size: 106 KB - Last synced at: 9 days ago - Pushed at: about 2 months ago - Stars: 31 - Forks: 5

google-deepmind/predictingthepast

Language: Python - Size: 1020 KB - Last synced at: 17 days ago - Pushed at: 17 days ago - Stars: 135 - Forks: 15

google-deepmind/forest_typology

Datasets to protect Earth's forests and biodiversity

Language: Jupyter Notebook - Size: 2.95 MB - Last synced at: 15 days ago - Pushed at: 4 months ago - Stars: 88 - Forks: 12

google-deepmind/recurrentgemma

Open weights language model from Google DeepMind, based on Griffin.

Language: Python - Size: 102 KB - Last synced at: 16 days ago - Pushed at: 3 months ago - Stars: 647 - Forks: 32

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: 766 KB - Last synced at: 18 days ago - Pushed at: about 1 month ago - Stars: 637 - Forks: 76

google-deepmind/tf2jax

Language: Python - Size: 440 KB - Last synced at: 19 days ago - Pushed at: 19 days ago - Stars: 115 - Forks: 15

google-deepmind/envlogger

A tool for recording RL trajectories.

Language: Python - Size: 474 KB - Last synced at: 6 days ago - Pushed at: about 1 month ago - Stars: 107 - Forks: 15

google-deepmind/simply

Language: Python - Size: 1.28 MB - Last synced at: 20 days ago - Pushed at: 20 days ago - Stars: 4 - Forks: 1

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,136 - Forks: 386

google-deepmind/surface-distance

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

Language: Python - Size: 39.1 KB - Last synced at: 7 days ago - Pushed at: 7 months ago - Stars: 576 - Forks: 120

google-deepmind/mathematics_conjectures

Language: Jupyter Notebook - Size: 34.2 KB - Last synced at: 13 days ago - Pushed at: almost 4 years ago - Stars: 377 - Forks: 43

google-deepmind/space_is_a_latent_sequence

Language: Jupyter Notebook - Size: 1.44 MB - Last synced at: 23 days ago - Pushed at: 23 days ago - Stars: 2 - Forks: 0

google-deepmind/neptune

Size: 2.06 MB - Last synced at: 20 days ago - Pushed at: 4 months ago - Stars: 70 - Forks: 9

google-deepmind/semppl 📦

Language: Python - Size: 32.2 KB - Last synced at: 14 days ago - Pushed at: over 2 years ago - Stars: 10 - Forks: 1

google-deepmind/rc-data 📦

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

Language: Python - Size: 1.26 MB - Last synced at: 26 days ago - Pushed at: over 8 years ago - Stars: 1,296 - Forks: 245

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: 22 days ago - Pushed at: almost 6 years ago - Stars: 662 - Forks: 124

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: 26 days ago - Pushed at: over 6 years ago - Stars: 1,007 - Forks: 160

google-deepmind/hanabi-learning-environment 📦

hanabi_learning_environment is a research platform for Hanabi experiments.

Language: Python - Size: 137 KB - Last synced at: 26 days ago - Pushed at: over 2 years ago - Stars: 654 - Forks: 157

google-deepmind/dm_memorytasks

A set of 13 diverse machine-learning tasks that require memory to solve.

Language: Python - Size: 27.3 KB - Last synced at: 19 days ago - Pushed at: about 4 years ago - Stars: 223 - Forks: 18

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: 5 days ago - Pushed at: over 2 years ago - Stars: 75 - Forks: 21

google-deepmind/dm_env_rpc

A networking protocol for agent-environment communication

Language: Python - Size: 406 KB - Last synced at: 19 days ago - Pushed at: 7 months ago - Stars: 102 - Forks: 14

google-deepmind/dnc

A TensorFlow implementation of the Differentiable Neural Computer.

Language: Python - Size: 71.3 KB - Last synced at: 26 days ago - Pushed at: about 4 years ago - Stars: 2,516 - Forks: 443

google-deepmind/fusion_surrogates

A library of surrogate transport models for tokamak fusion.

Language: Python - Size: 566 KB - Last synced at: 17 days ago - Pushed at: 17 days ago - Stars: 11 - Forks: 1

google-deepmind/alphafold3

AlphaFold 3 inference pipeline.

Language: Python - Size: 40 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 6,835 - Forks: 900

google-deepmind/c3_neural_compression

Language: Python - Size: 114 KB - Last synced at: 26 days ago - Pushed at: 3 months ago - Stars: 84 - Forks: 7

google-deepmind/icml2024-roundtrip-correctness

Language: Python - Size: 60.5 KB - Last synced at: 7 days ago - Pushed at: about 1 year ago - Stars: 13 - Forks: 4

google-deepmind/einshape

Language: Python - Size: 67.4 KB - Last synced at: 12 days ago - Pushed at: about 1 year ago - Stars: 105 - Forks: 8

google-deepmind/game_arena

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

google-deepmind/functa

Language: Python - Size: 52.8 MB - Last synced at: 3 days ago - Pushed at: about 1 year ago - Stars: 158 - Forks: 8

google-deepmind/anthro-benchmark

Language: Python - Size: 153 KB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 6 - Forks: 2

google-deepmind/logical-entailment-dataset 📦

This repository contains an entailment dataset for propositional logic, and code for generating that dataset. It also contains code for parsing the dataset in python.

Language: Haskell - Size: 1.86 MB - Last synced at: 19 days ago - Pushed at: over 7 years ago - Stars: 70 - Forks: 18

google-deepmind/alphaevolve_results

Language: Jupyter Notebook - Size: 3.79 MB - Last synced at: about 1 month ago - Pushed at: 3 months ago - Stars: 213 - Forks: 21

google-deepmind/proactive_t2i_agents

Code release for the paper, "Proactive Agents for Text-to-Image Generation under Uncertainty"

Language: Python - Size: 15.2 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 42 - Forks: 6

google-deepmind/leo

Implementation of Meta-Learning with Latent Embedding Optimization

Language: Python - Size: 25.4 KB - Last synced at: 4 days ago - Pushed at: over 6 years ago - Stars: 310 - Forks: 60

google-deepmind/multiscope

Real-time visualisation

Language: Go - Size: 2.16 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 17 - Forks: 4

google-deepmind/dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods

Language: Jupyter Notebook - Size: 28.2 MB - Last synced at: 13 days ago - Pushed at: over 4 years ago - Stars: 510 - Forks: 69

google-deepmind/jaxline

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

google-deepmind/tapnet

Tracking Any Point (TAP)

Language: Jupyter Notebook - Size: 4.92 MB - Last synced at: about 1 month ago - Pushed at: about 1 month ago - Stars: 1,586 - Forks: 150

google-deepmind/aloha_sim

A collection of tabletop tasks in Mujoco

Language: Python - Size: 50.2 MB - Last synced at: about 1 month ago - Pushed at: 2 months ago - Stars: 208 - Forks: 19

google-deepmind/levintreesearch_cm

Language: Racket - Size: 16.2 MB - Last synced at: 5 days ago - Pushed at: about 2 months ago - Stars: 32 - Forks: 5

google-deepmind/labmaze

A standalone release of DeepMind Lab's maze generator with Python bindings.

Language: C++ - Size: 4.2 MB - Last synced at: 19 days ago - Pushed at: almost 2 years ago - Stars: 64 - Forks: 20