← all repositories

Kaixhin/Rainbow

A PyTorch implementation of Rainbow DQN combining seven deep reinforcement learning improvements for Atari game agents.

1.7k stars Python ML Frameworks
Rainbow
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository implements the Rainbow algorithm, which integrates seven key improvements to Deep Q-Networks: Double DQN, Prioritised Experience Replay, Dueling Networks, Multi-step Returns, Distributional RL, and Noisy Nets. It uses PyTorch as the deep learning framework and is designed to train reinforcement learning agents on Atari games. The project includes both the original Rainbow configuration and a data-efficient variant.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.