18 projects
dm-haiku
Haiku is a library for building neural networks in JAX.
dm-env-rpc
A networking protocol for agent-environment communication.
optax
A gradient processing and optimisation library in JAX.
chex
Chex: Testing made fun, in JAX!
dm-control
Continuous control environments and MuJoCo Python bindings.
labmaze
LabMaze: DeepMind Lab's text maze generator.
open-spiel
A Framework for Reinforcement Learning in Games
bsuite
Core RL Behaviour Suite. A collection of reinforcement learning experiments.
dm-env
A Python interface for Reinforcement Learning environments.
fancyflags
A Python library for defining dictionary-valued flags.
rlax
A library of reinforcement learning building blocks in JAX.
dm-acme
A Python library for Reinforcement Learning.
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.
dm-sonnet
Sonnet is a library for building neural networks in TensorFlow.
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