SocialJax: sequential social dilemma environments

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

Code repository for "N-agent Ad Hoc Teamwork" paper (Wang et al., Neurips 2024).

A benchmark environment for fully cooperative human-AI performance.

Codebase for BRDiv: Diverse teammate generation for ad hoc teamwork

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

Adaptable Agent Populations via a Generative Model of Policies

Mind-aware Multi-agent Management Reinforcement Learning

GraphRNN: Generating Realistic Graphs with Deep Auto-regressive Models

Reproduces the results of MinCutPool as presented in the 2020 ICML paper "Spectral Clustering with Graph Neural Networks for Graph Pooling".

Deep Variational Reinforcement Learning

Library for Model Based RL

Level-based Foraging (LBF): A multi-agent environment for RL

Gradient based Hyperparameter Tuning library in PyTorch

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

PyTorch Implementation of MADDPG (Lowe et. al. 2017)