Discover google-deepmind/treescope Open Source project

Stars
256
Rank 159,219 (Top 4 %)
Language
Python
License
Apache License 2.0
Created 4 months ago
Updated 2 months ago

google-deepmind

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

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

deepmind-research

This repository contains implementations and illustrative code to accompany DeepMind publications

Jupyter Notebook

13,132

alphafold

Open source code for AlphaFold.

Python

12,602

sonnet

TensorFlow-based neural network library

Python

9,769

mujoco

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

Jupyter Notebook

8,113

pysc2

StarCraft II Learning Environment

Python

8,001

lab

A customisable 3D platform for agent-based AI research

7,101

graph_nets

Build Graph Nets in Tensorflow

graphcast

open_spiel

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

alphageometry

learning-to-learn

Learning to Learn in TensorFlow

Python

4,064

dm_control

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

Python

3,793

acme

A library of reinforcement learning components and agents

Python

3,466

trfl

TensorFlow Reinforcement Learning

Python

3,136

dm-haiku

JAX-based neural network library

alphatensor

dnc

A TensorFlow implementation of the Differentiable Neural Computer.

Python

2,478

gemma

Open weights LLM from Google DeepMind.

Python

2,421

mctx

Monte Carlo tree search in JAX

code_contests

optax

Optax is a gradient processing and optimization library for JAX.

Python

1,670

kinetics-i3d

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

Python

1,639

penzai

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

Python

1,639

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

bsuite

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

educational

jraph

A Graph Neural Network Library in Jax

Python

1,349

rc-data

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

Python

1,285

mujoco_menagerie

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

Jupyter Notebook

1,278

tapnet

Tracking Any Point (TAP)

rlax

scalable_agent

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

Python

981

android_env

RL research on Android devices.

Python

977

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

mujoco_mpc

Real-time behaviour synthesis with MuJoCo, using Predictive Control

C++

959

dramatron

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

Jupyter Notebook

947

tree

tree is a library for working with nested data structures

materials_discovery

xmanager

A platform for managing machine learning experiments

open_x_embodiment

chex

ferminet

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

Python

707

reverb

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

funsearch

alphadev

pycolab

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

Python

659

concordia

A library for generative social simulation

Python

634

hanabi-learning-environment

hanabi_learning_environment is a research platform for Hanabi experiments.

Python

614

recurrentgemma

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

Python

603

ai-safety-gridworlds

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

Python

577

meltingpot

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

Python

576

ithaca

Restoring and attributing ancient texts using deep neural networks

Jupyter Notebook

547

dqn

Lua/Torch implementation of DQN (Nature, 2015)

Lua

546

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).

distrax

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

surface-distance

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

tracr

alphamissense

dsprites-dataset

Dataset to assess the disentanglement properties of unsupervised learning methods

Jupyter Notebook

476

narrativeqa

This repository contains the NarrativeQA dataset. It includes the list of documents with Wikipedia summaries, links to full stories, and questions and answers.

clrs

lab2d

A customisable 2D platform for agent-based AI research

C++

420

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.

alphastar

dm_pix

PIX is an image processing library in JAX, for JAX.

Python

386

opro

official code for "Large Language Models as Optimizers"

Python

383

mathematics_conjectures

Jupyter Notebook

367

spriteworld

Spriteworld: a flexible, configurable python-based reinforcement learning environment

Python

367

torax

TORAX: Tokamak transport simulation in JAX

Python

361

dm_env

A Python interface for reinforcement learning environments

Python

343

dm_robotics

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

Python

341

spiral

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

launchpad

leo

Implementation of Meta-Learning with Latent Embedding Optimization

enn

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

gqn-datasets

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

Python

269

multi_object_datasets

Multi-object image datasets with ground-truth segmentation masks and generative factors.

Python

254

AQuA

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

238

synjax

Python

238

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

card2code

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

236

arnheim

Jupyter Notebook

235

torch-hdf5

Torch interface to HDF5 library

Lua

234

kfac-jax

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

Python

231

dm_memorytasks

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

Python

221

Temporal-3D-Pose-Kinetics

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

Python

218

dm_alchemy

DeepMind Alchemy task environment: a meta-reinforcement learning benchmark

neural_testbed

perception_test

jmp

JMP is a Mixed Precision library for JAX.

Python

183

neural_networks_chomsky_hierarchy

Neural Networks and the Chomsky Hierarchy

xquad

nanodo

pg19

spectral_inference_networks

Implementation of Spectral Inference Networks, ICLR 2019

Python

165

barkour_robot

Barkour Robot: Agile Quadruped Robots by Google DeepMind

onetwo

abstract-reasoning-matrices

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

162

google-deepmind/treescope

google-deepmind

Reviews

Repository Details

More Repositories