backpropper

acme

A library of reinforcement learning components and agents

alphafold

Open source code for AlphaFold.

bsuite

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

coinrun

Code for the paper "Quantifying Transfer in Reinforcement Learning"

deep-rl-class

This repo contain the syllabus of the Hugging Face Deep Reinforcement Learning Class.

dm-haiku

JAX-based neural network library

dm_control

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

dopamine

Dopamine is a research framework for fast prototyping of reinforcement learning algorithms.

enn

flax

Flax is a neural network library for JAX that is designed for flexibility.

football

Check out the new game server:

gym-minigrid

Minimalistic gridworld package for OpenAI Gym

idaac

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

KAT

Research code for "KAT: A Knowledge Augmented Transformer for Vision-and-Language"

lab2d

A customisable 2D platform for agent-based AI research

mctx

Monte Carlo tree search in JAX

meltingpot

mujoco

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

multi-agent-emergence-environments

Environment generation code for the paper "Emergent Tool Use From Multi-Agent Autocurricula"

nle

The NetHack Learning Environment

objax

ray

An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.

rl-baselines3-zoo

A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.

rlax

rliable

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.

rlpyt

Reinforcement Learning in PyTorch

sequential_social_dilemma_games

Repo for reproduction of sequential social dilemmas

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

xmanager

A platform for managing machine learning experiments