A toolkit for reproducible reinforcement learning research.

official repo for paper Unsupervised Skill Discovery via Recurrent Skill Training

Related papers for reinforcement learning, including classic papers and latest papers in top conferences

PyTorch implementation of the discrete Soft-Actor-Critic algorithm.

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch

An OpenAI Gym interface to The Legend of Zelda on the NES.

Playing Pokemon Red with Reinforcement Learning

Probability and Computing (Second Edition). Michael Mitzenmacher, Eli Upfal

Implementation of PGA-MAP-Elites for the TELO paper.

Repository for the QDgym code. A framework for Quality Diversity optimization benchmark tasks based OpenAI Gym.

Repository for the PGA-MAP-Elites algorithm. PGA-MAP-Elites was developed to efficiently scale MAP-Elites to large genotypes and noisy domains. It uses Neuroevolution driven by a Genetic Algorithm …

Soft Actor-Critic

Paired Open-Ended Trailblazer (POET) and Enhanced POET

Stable Diffusion web UI

A latent text-to-image diffusion model

Implementation of Vision Transformer, a simple way to achieve SOTA in vision classification with only a single transformer encoder, in Pytorch

PyTorch implementation of FQF, IQN and QR-DQN.

FQF(Fully parameterized Quantile Function for distributional reinforcement learning) is a general reinforcement learning framework for Atari games, which can learn to play Atari games automatically…

Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"

MineRL Competition for Sample Efficient Reinforcement Learning - Python Package

SpotTune: Transfer Learning through Adaptive Fine-tuning

PyTorch Re-Implementation of "The Sparsely-Gated Mixture-of-Experts Layer" by Noam Shazeer et al. https://arxiv.org/abs/1701.06538

Monte Carlo tree search in JAX

Project Malmo is a platform for Artificial Intelligence experimentation and research built on top of Minecraft. We aim to inspire a new generation of research into challenging new problems presente…

Codebase for the solution that won first place and was awarded the most human-like agent in the 2021 NeurIPS Competition MineRL BASALT Challenge.

⭐️ Companies that don't have a broken hiring process

Styled text-to-drawing synthesis method. Featured at IJCAI 2022 and the 2021 NeurIPS Workshop on Machine Learning for Creativity and Design

Structured illumination microscopy in python

An educational resource to help anyone learn deep reinforcement learning.

Unofficial Alias-Free GAN implementation. Based on rosinality's version with expanded training and inference options.