← all repositories

kzl/decision-transformer

PyTorch implementation of Decision Transformer, a reinforcement learning model that reformulates RL as autoregressive sequence modeling.

2.8k stars Python ML FrameworksAgents
decision-transformer
Velocity · 7d
+1.5
★ / day
Trend
steady
star history

The repository provides the official implementation of Decision Transformer, a 2021 paper that treats reinforcement learning as a sequence modeling problem. It uses a transformer architecture to predict actions based on past states, returns, and actions, enabling offline RL without explicit value estimation. The codebase includes scripts for Atari games and OpenAI Gym environments to reproduce the paper’s experimental results.

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