40 projects
dm-reverb-nightly
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research.
xmanager
A framework for managing machine learning experiments
dm-reverb
Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research.
dm-pix
PIX is an image processing library in JAX, for JAX.
dm-launchpad
Launchpad is a library that simplifies writing distributed programs and seamlessly launching them on a range of supported platforms.
dm-launchpad-nightly
Launchpad is a library that simplifies writing distributed programs and seamlessly launching them on a range of supported platforms.
envlogger
EnvLogger: A tool for recording trajectories.
chex
Chex: Testing made fun, in JAX!
dm-control
Continuous control environments and MuJoCo Python bindings.
mujoco
MuJoCo Physics Simulator
optax
A gradient processing and optimisation library in JAX.
dm-tree
Tree is a library for working with nested data structures.
distrax
Distrax: Probability distributions in JAX.
open-spiel
A Framework for Reinforcement Learning in Games
rlax
A library of reinforcement learning building blocks in JAX.
dm-haiku
Haiku is a library for building neural networks in JAX.
dm-acme
A Python library for Reinforcement Learning.
dm-robotics-transformations
A library for Rigid Body Transformations
dm-robotics-moma
Tools for authoring robotic manipulation tasks.
dm-robotics-manipulation
Parametrically defined mesh objects
dm-robotics-geometry
This library provides primitives for dealing with scene and robot geometry
dm-robotics-controllers
Python bindings for dm_robotics/cpp/controllers
dm-robotics-agentflow
Tools for single-embodiment, multiple-task, Reinforcement Learning
dm-env-rpc
A networking protocol for agent-environment communication.
fancyflags
A Python library for defining dictionary-valued flags.
jraph
Jraph: A library for Graph Neural Networks in Jax
android-env
AndroidEnv environment and library for training agents.
dm-clrs
The CLRS Algorithmic Reasoning Benchmark.
trfl
trfl is a library of building blocks for reinforcement learning algorithms.
dm-memorytasks
DeepMind Memory Tasks, a set of Unity-based machine-learning research tasks.
labmaze
LabMaze: DeepMind Lab's text maze generator.
alphafold
An implementation of the inference pipeline of AlphaFold v2.0.This is a completely new model that was entered as AlphaFold2 in CASP14 and published in Nature.
dm-env
A Python interface for Reinforcement Learning environments.
jmp
JMP is a Mixed Precision library for JAX.
bsuite
Core RL Behaviour Suite. A collection of reinforcement learning experiments.
dm-sonnet
Sonnet is a library for building neural networks in TensorFlow.
graph-nets
Library for building graph networks in Tensorflow and Sonnet.
PySC2
Starcraft II environment and library for training agents.
dm-sonnet-gpu
Sonnet is a library for building neural networks in TensorFlow.
mathematics-dataset
A synthetic dataset of school-level mathematics questions