Google DeepMind
Google DeepMind
meltingpot
A suite of test scenarios for multi-agent reinforcement learning.
dsprites-dataset
Dataset to assess the disentanglement properties of unsupervised learning methods
open_spiel
OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.
mctx
Monte Carlo tree search in JAX
learning-to-learn
Learning to Learn in TensorFlow
dm_pix
PIX is an image processing library in JAX, for JAX.
dm_env
A Python interface for reinforcement learning environments
lab2d
A customisable 2D platform for agent-based AI research
distribution_shift_framework
This repository contains the code of the distribution shift framework presented in A Fine-Grained Analysis on Distribution Shift (Wiles et al., 2022).
envlogger
A tool for recording RL trajectories.